com.databricks.labs.automl.pipeline.HasFeaturesColumns.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.{Params, StringArrayParam}

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

  final val featureColumns: StringArrayParam = new StringArrayParam(this, "featureColumns", "List of feature column names")

  def setFeatureColumns(value: Array[String]): this.type = set(featureColumns, value)

  def getFeatureColumns: Array[String] = $(featureColumns)

}