• Home
• io.hydrosphere
• spark-ml-serving-2_0_2.11
• 0.3.2
• source code
• SpecificTransformerConversions.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.

io.hydrosphere.spark_ml_serving.SpecificTransformerConversions.scala Maven / Gradle / Ivy

package io.hydrosphere.spark_ml_serving

import io.hydrosphere.spark_ml_serving.classification._
import io.hydrosphere.spark_ml_serving.common.LocalTransformer
import io.hydrosphere.spark_ml_serving.preprocessors.LocalWord2VecModel
import org.apache.spark.ml.Transformer
import org.apache.spark.ml.classification._
import org.apache.spark.ml.feature.Word2VecModel

object SpecificTransformerConversions extends DynamicTransformerConverter {

  implicit def transformerToLocal(transformer: Transformer): LocalTransformer[_] = {
    transformer match {
      case x: LogisticRegressionModel => new LocalLogisticRegressionModel(x)
      case x: Word2VecModel           => new LocalWord2VecModel(x)

      case x => throw new Exception(s"Unknown model: ${x.getClass}")
    }
  }
}