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 Details

    • XGBoostClassificationModelConverter

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

  • Method Details

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