io.hydrosphere.mist.ml.loaders.LocalHashingTF.scala Maven / Gradle / Ivy
package io.hydrosphere.mist.ml.loaders
import io.hydrosphere.mist.ml.Metadata
import org.apache.spark.ml.Transformer
import org.apache.spark.ml.feature.HashingTF
object LocalHashingTF extends LocalModel {
override def localLoad(metadata: Metadata, data: Map[String, Any]): Transformer = {
new HashingTF(metadata.uid)
.setInputCol(metadata.paramMap("inputCol").asInstanceOf[String])
.setOutputCol(metadata.paramMap("outputCol").asInstanceOf[String])
.setBinary(metadata.paramMap("binary").asInstanceOf[Boolean])
.setNumFeatures(metadata.paramMap("numFeatures").asInstanceOf[Int])
}
}