org.jpmml.sparkml

Class PredictionModelConverter<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.PredictionModelConverter<T>

All Implemented Interfaces:

org.jpmml.converter.HasOptions, HasSparkMLOptions, HasPredictionModelOptions

Direct Known Subclasses:

ClassificationModelConverter, RegressionModelConverter

public abstract class PredictionModelConverter<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>
implements HasPredictionModelOptions

Field Summary

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

OPTION_KEEP_PREDICTIONCOL

Constructor Summary

Constructors

Constructor and Description
PredictionModelConverter(T model)

Method Summary

Modifier and Type	Method and Description
List<org.jpmml.converter.Feature>	getFeatures(SparkMLEncoder encoder)
org.jpmml.converter.Label	getLabel(SparkMLEncoder encoder)

Methods inherited from class org.jpmml.sparkml.ModelConverter

checkSchema, encodeModel, encodeSchema, getMiningFunction, getModel, registerModel, registerOutputFields

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

PredictionModelConverter

public PredictionModelConverter(T model)

Method Detail

getLabel

public org.jpmml.converter.Label getLabel(SparkMLEncoder encoder)

Overrides:

getLabel 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>

getFeatures

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

Specified by:

getFeatures 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>