xgboost.sklearn

Class BoosterUtil

java.lang.Object

xgboost.sklearn.BoosterUtil

public class BoosterUtil
extends Object

Method Summary

Modifier and Type	Method and Description
static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.dmg.pmml.mining.MiningModel	encodeModel(E estimator, org.jpmml.converter.Schema schema)
static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.dmg.pmml.PMML	encodePMML(E estimator)
static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> int	getNumberOfFeatures(E estimator)
static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.jpmml.xgboost.ObjFunction	getObjFunction(E estimator)

Methods inherited from class java.lang.Object

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

Method Detail

getNumberOfFeatures

public static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> int getNumberOfFeatures(E estimator)

getObjFunction

public static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.jpmml.xgboost.ObjFunction getObjFunction(E estimator)

encodeModel

public static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.dmg.pmml.mining.MiningModel encodeModel(E estimator,
                                                                                                                                   org.jpmml.converter.Schema schema)

encodePMML

public static <E extends Estimator & HasBooster & org.jpmml.xgboost.HasXGBoostOptions> org.dmg.pmml.PMML encodePMML(E estimator)