Package 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.HasOptions, org.jpmml.sparkml.HasSparkMLOptions, org.jpmml.sparkml.model.HasPredictionModelOptions
public class XGBoostRegressionModelConverter extends org.jpmml.sparkml.PredictionModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>

  • Field Summary

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

    OPTION_KEEP_PREDICTIONCOL

  • Constructor Summary

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

  • Method Summary

    Modifier and Type
    Method
    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, org.jpmml.sparkml.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

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

  • Constructor Details

    • XGBoostRegressionModelConverter

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

  • Method Details

    • getMiningFunction

      public org.dmg.pmml.MiningFunction getMiningFunction()
      Specified by:
      getMiningFunction in class org.jpmml.sparkml.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 org.jpmml.sparkml.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, org.jpmml.sparkml.SparkMLEncoder encoder)
      Overrides:
      registerOutputFields in class org.jpmml.sparkml.ModelConverter<ml.dmlc.xgboost4j.scala.spark.XGBoostRegressionModel>