• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• AutoTrainer.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.microsoft.ml.spark.train.AutoTrainer.scala Maven / Gradle / Ivy

// Copyright (C) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in project root for information.

package com.microsoft.ml.spark.train

import com.microsoft.ml.spark.core.contracts.{HasFeaturesCol, HasLabelCol}
import org.apache.spark.ml.{ComplexParamsWritable, Estimator, Model}
import org.apache.spark.ml.param.{EstimatorParam, IntParam}

/** Defines common inheritance and parameters across trainers.
  */
trait AutoTrainer[TrainedModel <: Model[TrainedModel]] extends Estimator[TrainedModel]
  with HasLabelCol with ComplexParamsWritable with HasFeaturesCol {

  /** Doc for model to run.
    */
  def modelDoc: String

  /** Number of features to hash to
    * @group param
    */
  val numFeatures = new IntParam(this, "numFeatures", "Number of features to hash to")
  setDefault(numFeatures -> 0)

  /** @group getParam */
  def getNumFeatures: Int = $(numFeatures)
  /** @group setParam */
  def setNumFeatures(value: Int): this.type = set(numFeatures, value)

  /** Model to run.  See doc on derived classes.
    * @group param
    */
  val model = new EstimatorParam(this, "model", modelDoc)
  /** @group getParam */
  def getModel: Estimator[_ <: Model[_]] = $(model)
  /** @group setParam */
  def setModel(value: Estimator[_ <: Model[_]]): this.type = set(model, value)
}