org.jpmml.sparkml

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

Direct Known Subclasses:

DecisionTreeRegressionModelConverter, GBTRegressionModelConverter, GeneralizedLinearRegressionModelConverter, LinearRegressionModelConverter, RandomForestRegressionModelConverter

public abstract class RegressionModelConverter<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
RegressionModelConverter(T model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.Output	encodeOutput(org.jpmml.converter.Label label, SparkMLEncoder encoder)
org.dmg.pmml.MiningFunction	getMiningFunction()

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeModel, encodeSchema, registerModel

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

RegressionModelConverter

public RegressionModelConverter(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>

encodeOutput

public org.dmg.pmml.Output encodeOutput(org.jpmml.converter.Label label,
                                        SparkMLEncoder encoder)

Overrides:

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