override def featureDimension()

in core/src/main/scala/com/spotify/featran/transformers/TopNOneHotEncoder.scala [109:115]

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

  override def featureDimension(c: SortedMap[String, Int]): Int =
    if (encodeMissingValue) c.size + 1 else c.size

  override def featureNames(c: SortedMap[String, Int]): Seq[String] = {
    val names = c.iterator.map(name + '_' + _._1).toSeq
    if (encodeMissingValue) names :+ (name + '_' + MissingValueToken) else names
  }