com.databricks.labs.automl.pipeline.HasFeatureColumn.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of automatedml_2.11 Show documentation

Databricks Labs AutoML toolkit

The newest version!

package com.databricks.labs.automl.pipeline

import org.apache.spark.ml.param.{Param, Params}

/**
  * @author Jas Bali
  *
  */
trait HasFeatureColumn extends Params {

  final val featureCol: Param[String] = new Param[String](this, "featureCol", "Feature Column Name")

  def setFeatureCol(value: String): this.type = set(featureCol, value)

  def getFeatureCol: String = $(featureCol)
}