org.jpmml.sparkml.xgboost

Class XGBoostRegressionModelConverter

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

org.jpmml.sparkml.PredictionModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>

org.jpmml.sparkml.xgboost.XGBoostRegressionModelConverter

All Implemented Interfaces:

org.jpmml.converter.HasNativeConfiguration, org.jpmml.converter.HasOptions, HasSparkMLOptions, HasPredictionModelOptions, HasSparkMLXGBoostOptions, org.jpmml.xgboost.HasXGBoostOptions

public class XGBoostRegressionModelConverter
extends PredictionModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>
implements HasSparkMLXGBoostOptions

Field Summary

Fields inherited from interface org.jpmml.sparkml.xgboost.HasSparkMLXGBoostOptions

OPTION_INPUT_FLOAT

Fields inherited from interface org.jpmml.xgboost.HasXGBoostOptions

OPTION_BYTE_ORDER, OPTION_CHARSET, OPTION_COMPACT, OPTION_MISSING, OPTION_NTREE_LIMIT, OPTION_NUMERIC, OPTION_PRUNE

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

OPTION_KEEP_PREDICTIONCOL

Constructor Summary

Constructors

Constructor and Description
XGBoostRegressionModelConverter(ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.mining.MiningModel	encodeModel(org.jpmml.converter.Schema schema)
org.dmg.pmml.MiningFunction	getMiningFunction()
List<org.dmg.pmml.OutputField>	registerOutputFields(org.jpmml.converter.Label label, org.dmg.pmml.Model pmmlModel, SparkMLEncoder encoder)

Methods inherited from class org.jpmml.sparkml.PredictionModelConverter

getFeatures, getLabel

Methods inherited from class org.jpmml.sparkml.ModelConverter

checkSchema, 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.xgboost.HasXGBoostOptions

getNativeConfiguration

Constructor Detail

XGBoostRegressionModelConverter

public XGBoostRegressionModelConverter(ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel model)

Method Detail

getMiningFunction

public org.dmg.pmml.MiningFunction getMiningFunction()

Specified by:

getMiningFunction in class ModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>

encodeModel

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

Specified by:

encodeModel in class ModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>

registerOutputFields

public List<org.dmg.pmml.OutputField> registerOutputFields(org.jpmml.converter.Label label,
                                                           org.dmg.pmml.Model pmmlModel,
                                                           SparkMLEncoder encoder)

Overrides:

registerOutputFields in class ModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>