def apply()

in core/src/main/scala/com/spotify/featran/transformers/TopNOneHotEncoder.scala [55:80]

• 19 lines of code
• 1 McCabe index (conditional complexity)

  def apply(
    name: String,
    n: Int,
    eps: Double = 0.001,
    delta: Double = 0.001,
    seed: Int = Random.nextInt(),
    encodeMissingValue: Boolean = false
  ): Transformer[String, SketchMap[String, Long], SortedMap[String, Int]] =
    new TopNOneHotEncoder(name, n, eps, delta, seed, encodeMissingValue)

  /**
   * Create a new [[TopNOneHotEncoder$]] from a settings object
   * @param setting
   *   Settings object
   */
  def fromSettings(
    setting: Settings
  ): Transformer[String, SketchMap[String, Long], SortedMap[String, Int]] = {
    val n = setting.params("n").toInt
    val eps = setting.params("eps").toDouble
    val delta = setting.params("delta").toDouble
    val seed = setting.params("seed").toInt
    val encodeMissingValue = setting.params("encodeMissingValue").toBoolean

    TopNOneHotEncoder(setting.name, n, eps, delta, seed, encodeMissingValue)
  }