org.jpmml.sparkml

Class ClassificationModelConverter<T extends org.apache.spark.ml.PredictionModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasLabelCol & 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, LinearSVCModelConverter, LogisticRegressionModelConverter, MultilayerPerceptronClassificationModelConverter, NaiveBayesModelConverter, RandomForestClassificationModelConverter

public abstract class ClassificationModelConverter<T extends org.apache.spark.ml.PredictionModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasLabelCol & 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 model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.MiningFunction	getMiningFunction()
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.ModelConverter

encodeModel, encodeSchema, registerModel

Methods inherited from class org.jpmml.sparkml.TransformerConverter

getOption, getOptions, getTransformer, setOptions

Methods inherited from class java.lang.Object

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

Constructor Detail

ClassificationModelConverter

public ClassificationModelConverter(T model)

Method Detail

getMiningFunction

public org.dmg.pmml.MiningFunction getMiningFunction()

Specified by:

getMiningFunction in class ModelConverter<T extends org.apache.spark.ml.PredictionModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasLabelCol & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>

registerOutputFields

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

Overrides:

registerOutputFields in class ModelConverter<T extends org.apache.spark.ml.PredictionModel<org.apache.spark.ml.linalg.Vector,T> & org.apache.spark.ml.param.shared.HasLabelCol & org.apache.spark.ml.param.shared.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>