• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• AutoTrainedModel.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.AutoTrainedModel.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.serialize.ConstructorWritable
import org.apache.spark.ml.{Model, PipelineModel, Transformer}
import org.apache.spark.ml.param.ParamMap

/** Defines common inheritance and functions across auto trained models.
  */
abstract class AutoTrainedModel[TrainedModel <: Model[TrainedModel]](val model: PipelineModel)
  extends Model[TrainedModel] with ConstructorWritable[TrainedModel] {
  /** Retrieve the param map from the underlying model.
    * @return The param map from the underlying model.
    */
  def getParamMap: ParamMap = model.stages.last.extractParamMap()

  /** Retrieve the underlying model.
    * @return The underlying model.
    */
  def getModel: Transformer = model.stages.last
}