Uses of Enum Class
org.tribuo.regression.xgboost.XGBoostRegressionTrainer.RegressionType
Package
Description
Provides an interface to XGBoost for regression problems.
-
Uses of XGBoostRegressionTrainer.RegressionType in org.tribuo.regression.xgboost
Modifier and TypeFieldDescriptionTrainTest.XGBoostOptions.rType
Regression type to use.XGBoostOptions.rType
Regression type to use.Modifier and TypeMethodDescriptionReturns the enum constant of this class with the specified name.XGBoostRegressionTrainer.RegressionType.values()
Returns an array containing the constants of this enum class, in the order they are declared.ModifierConstructorDescriptionXGBoostRegressionTrainer
(XGBoostTrainer.BoosterType boosterType, XGBoostTrainer.TreeMethod treeMethod, XGBoostRegressionTrainer.RegressionType rType, int numTrees, double eta, double gamma, int maxDepth, double minChildWeight, double subsample, double featureSubsample, double lambda, double alpha, int nThread, XGBoostTrainer.LoggingVerbosity verbosity, long seed) Create an XGBoost trainer.XGBoostRegressionTrainer
(XGBoostRegressionTrainer.RegressionType rType, int numTrees) Creates an XGBoostRegressionTrainer using the default parameters, the supplied loss and the supplied number of trees.XGBoostRegressionTrainer
(XGBoostRegressionTrainer.RegressionType rType, int numTrees, double eta, double gamma, int maxDepth, double minChildWeight, double subsample, double featureSubsample, double lambda, double alpha, int nThread, boolean silent, long seed) Create an XGBoost trainer.XGBoostRegressionTrainer
(XGBoostRegressionTrainer.RegressionType rType, int numTrees, int numThreads, boolean silent) Creates an XGBoostRegressionTrainer using the default parameters with the supplied loss, number of trees, number of threads, and logging level.XGBoostRegressionTrainer
(XGBoostRegressionTrainer.RegressionType rType, int numTrees, Map<String, Object> parameters) This gives direct access to the XGBoost parameter map.