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

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.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>
extends ModelConverter<T>

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)

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeModel, encodeSchema, getLabel, getMiningFunction, registerModel, registerOutputFields

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

PredictionModelConverter

public PredictionModelConverter(T model)

Method Detail

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.HasFeaturesCol & org.apache.spark.ml.param.shared.HasPredictionCol>