org.jpmml.sparkml

Class ProbabilisticClassificationModelConverter<T extends org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasProbabilityCol>

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

org.jpmml.sparkml.PredictionModelConverter<T>

org.jpmml.sparkml.ClassificationModelConverter<T>

org.jpmml.sparkml.ProbabilisticClassificationModelConverter<T>

All Implemented Interfaces:

org.jpmml.converter.HasOptions, HasSparkMLOptions, HasPredictionModelOptions

Direct Known Subclasses:

DecisionTreeClassificationModelConverter, GBTClassificationModelConverter, LogisticRegressionModelConverter, MultilayerPerceptronClassificationModelConverter, NaiveBayesModelConverter, RandomForestClassificationModelConverter

public abstract class ProbabilisticClassificationModelConverter<T extends org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasProbabilityCol>
extends ClassificationModelConverter<T>

Field Summary

Fields inherited from interface org.jpmml.sparkml.model.HasPredictionModelOptions

OPTION_KEEP_PREDICTIONCOL

Constructor Summary

Constructors

Constructor and Description
ProbabilisticClassificationModelConverter(T model)

Method Summary

Modifier and Type	Method and Description
List<org.dmg.pmml.OutputField>	registerOutputFields(org.jpmml.converter.Label label, org.dmg.pmml.Model pmmlModel, SparkMLEncoder encoder)

Methods inherited from class org.jpmml.sparkml.ClassificationModelConverter

checkSchema, getMiningFunction, getNumberOfClasses

Methods inherited from class org.jpmml.sparkml.PredictionModelConverter

getFeatures, getLabel

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeModel, encodeSchema, getModel, registerModel

Methods inherited from class org.jpmml.sparkml.TransformerConverter

getObject, getOption, getOptions, setOptions

Methods inherited from class java.lang.Object

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

Constructor Detail

ProbabilisticClassificationModelConverter

public ProbabilisticClassificationModelConverter(T model)

Method Detail

registerOutputFields

public List<org.dmg.pmml.OutputField> registerOutputFields(org.jpmml.converter.Label label,
                                                           org.dmg.pmml.Model pmmlModel,
                                                           SparkMLEncoder encoder)

Overrides:

registerOutputFields in class ClassificationModelConverter<T extends org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasProbabilityCol>