io.github.metarank.ltrlib.model.DatasetDescriptor.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ltrlib_3 Show documentation

ltrlib

There is a newer version: 0.2.6

package io.github.metarank.ltrlib.model

case class DatasetDescriptor(offsets: Map[Feature, Int], features: List[Feature], dim: Int) {}

object DatasetDescriptor {
  def apply(features: List[Feature]) = {
    val offsets = features
      .map(_.size)
      .scanLeft(0)(_ + _)
      .zip(features)
      .map(_.swap)
    val nonUnique = features.map(_.name).groupBy(identity).filter(_._2.size > 1).keys.toList
    if (nonUnique.nonEmpty) throw new IllegalArgumentException(s"non-unique feature names: $nonUnique")
    new DatasetDescriptor(
      offsets = offsets.toMap,
      features = features,
      dim = features.map(_.size).sum
    )
  }
}