org.jpmml.sparkml.model

Class GBTClassificationModelConverter

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<org.apache.spark.ml.classification.GBTClassificationModel>

org.jpmml.sparkml.model.GBTClassificationModelConverter

All Implemented Interfaces:

org.jpmml.converter.HasNativeConfiguration, org.jpmml.converter.HasOptions, HasSparkMLOptions, HasFeatureImportances, HasPredictionModelOptions, HasTreeOptions

public class GBTClassificationModelConverter
extends ProbabilisticClassificationModelConverter<org.apache.spark.ml.classification.GBTClassificationModel>
implements HasFeatureImportances, HasTreeOptions

Field Summary

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

OPTION_COMPACT, OPTION_ESTIMATE_FEATURE_IMPORTANCES

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

OPTION_KEEP_PREDICTIONCOL

Constructor Summary

Constructors

Constructor and Description
GBTClassificationModelConverter(org.apache.spark.ml.classification.GBTClassificationModel model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.mining.MiningModel	encodeModel(org.jpmml.converter.Schema schema)
org.apache.spark.ml.linalg.Vector	getFeatureImportances()

Methods inherited from class org.jpmml.sparkml.ProbabilisticClassificationModelConverter

registerOutputFields

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

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

Methods inherited from interface org.jpmml.sparkml.model.HasTreeOptions

getNativeConfiguration

Constructor Detail

GBTClassificationModelConverter

public GBTClassificationModelConverter(org.apache.spark.ml.classification.GBTClassificationModel model)

Method Detail

getFeatureImportances

public org.apache.spark.ml.linalg.Vector getFeatureImportances()

Specified by:

getFeatureImportances in interface HasFeatureImportances

encodeModel

public org.dmg.pmml.mining.MiningModel encodeModel(org.jpmml.converter.Schema schema)

Specified by:

encodeModel in class ModelConverter<org.apache.spark.ml.classification.GBTClassificationModel>