org.jpmml.sparkml

Class ClassificationModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

org.jpmml.sparkml.ClassificationModelConverter<T>

Direct Known Subclasses:

DecisionTreeClassificationModelConverter, GBTClassificationModelConverter, LogisticRegressionModelConverter, MultilayerPerceptronClassificationModelConverter, RandomForestClassificationModelConverter

public abstract class ClassificationModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>
extends ModelConverter<T>

Constructor Summary

Constructors

Constructor and Description
ClassificationModelConverter(T transformer)

Method Summary

Methods

Modifier and Type	Method and Description
List<org.jpmml.converter.Feature>	encodeFeatures(SparkMLEncoder encoder)
org.dmg.pmml.MiningFunction	getMiningFunction()

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeModel

Methods inherited from class org.jpmml.sparkml.TransformerConverter

getTransformer

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ClassificationModelConverter

public ClassificationModelConverter(T transformer)

Method Detail

getMiningFunction

public org.dmg.pmml.MiningFunction getMiningFunction()

Specified by:

getMiningFunction in class ModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>

encodeFeatures

public List<org.jpmml.converter.Feature> encodeFeatures(SparkMLEncoder encoder)

Specified by:

encodeFeatures in class TransformerConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>