• Home
• com.databricks.labs
• automl-toolkit
• 0.8.1
• source code
• PipelineEnums.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

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

package com.databricks.labs.automl.pipeline

import com.databricks.labs.automl.params.{MLFlowConfig, MainConfig}

object PipelineEnums extends Enumeration {

  type PipelineEnums = PipelineConstants

  val OHE_SUFFIX = PipelineConstants("_oh")
  val SI_SUFFIX = PipelineConstants("_si")
  val FEATURE_NAME_TEMP_SUFFIX = PipelineConstants("_r")

  val LABEL_STRING_INDEXER_STAGE_NAME = PipelineConstants("LabelStringIndexer")

  case class PipelineConstants(value: String) extends Val
}


object PipelineVars extends Enumeration {

  type PipelineVars = PipelineVarsPair

  val PIPELINE_LABEL_REFACTOR_NEEDED_KEY = PipelineVarsPair("labelRefactorNeeded", classOf[Boolean])
  val MLFLOW_RUN_ID = PipelineVarsPair("MlFlowRunId", classOf[String])
  val MAIN_CONFIG = PipelineVarsPair("MainConfig", classOf[MainConfig])
  val PIPELINE_STATUS = PipelineVarsPair("PipelineStatus", classOf[String])
  val KSAMPLER_STAGES = PipelineVarsPair("KSamplerStages", classOf[String])

  case class PipelineVarsPair(key: String, keyType: Class[_]) extends Val
}