Package org.jpmml.sparkml.xgboost

Class XGBoostClassificationModelConverter

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<ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel>

org.jpmml.sparkml.xgboost.XGBoostClassificationModelConverter

All Implemented Interfaces:

org.jpmml.converter.HasOptions, org.jpmml.sparkml.HasSparkMLOptions, org.jpmml.sparkml.model.HasPredictionModelOptions

public class XGBoostClassificationModelConverter
extends org.jpmml.sparkml.ProbabilisticClassificationModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel>

Field Summary

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

OPTION_KEEP_PREDICTIONCOL

Constructor Summary

Constructors

Constructor	Description
XGBoostClassificationModelConverter(ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel model)

Method Summary

Modifier and Type	Method	Description
org.dmg.pmml.mining.MiningModel	encodeModel(org.jpmml.converter.Schema schema)
int	getNumberOfClasses()

Methods inherited from class org.jpmml.sparkml.ProbabilisticClassificationModelConverter

registerOutputFields

Methods inherited from class org.jpmml.sparkml.ClassificationModelConverter

checkSchema, getMiningFunction

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

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

Constructor Detail

XGBoostClassificationModelConverter

public XGBoostClassificationModelConverter(ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel model)

Method Detail

getNumberOfClasses

public int getNumberOfClasses()

Overrides:

getNumberOfClasses in class org.jpmml.sparkml.ClassificationModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel>

encodeModel

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

Specified by:

encodeModel in class org.jpmml.sparkml.ModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostClassificationModel>