• Home
• com.databricks.labs
• automl-toolkit
• 0.8.1
• source code
• TuningRunner.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.ensemble.tuner.TuningRunner.scala Maven / Gradle / Ivy

package com.databricks.labs.automl.ensemble.tuner

import com.databricks.labs.automl.model.tools.structures.TrainSplitReferences
import com.databricks.labs.automl.params.{DataGeneration, MainConfig, TunerOutput}

private[ensemble] object TuningRunner {

  def runTuning(mainConfig: MainConfig,
                payload: DataGeneration,
                testTrainSplitData: Array[TrainSplitReferences]): TunerOutput = {
    val tuner = ModelFamilyTunerType.getTunerInstanceByModelFamily(
      mainConfig.modelFamily,
      mainConfig,
      payload,
      testTrainSplitData
    )
    tuner.tune
  }



}