override def buildFeatures()

in core/src/main/scala/com/spotify/featran/transformers/TopNOneHotEncoder.scala [124:141]

• 18 lines of code
• 6 McCabe index (conditional complexity)

  override def buildFeatures(
    a: Option[String],
    c: SortedMap[String, Int],
    fb: FeatureBuilder[_]
  ): Unit = a match {
    case Some(k) =>
      c.get(k) match {
        case Some(v) =>
          fb.skip(v)
          fb.add(name + '_' + k, 1.0)
          fb.skip(math.max(0, c.size - v - 1))
          if (encodeMissingValue) fb.skip()
        case None =>
          addNonTopItem(c, fb)
          fb.reject(this, FeatureRejection.Unseen(Set(k)))
      }
    case None => addNonTopItem(c, fb)
  }