com.databricks.labs.automl.pipeline.HasLabelColumn.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 HasLabelColumn extends Params{

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

  def setLabelColumn(value: String): this.type = set(labelColumn, value)

  def getLabelColumn: String = $(labelColumn)
}