Package | Description |
---|---|
org.tribuo.regression.xgboost |
Provides an interface to XGBoost for regression problems.
|
Modifier and Type | Field and Description |
---|---|
XGBoostRegressionTrainer.RegressionType |
XGBoostOptions.rType |
XGBoostRegressionTrainer.RegressionType |
TrainTest.XGBoostOptions.rType |
Modifier and Type | Method and Description |
---|---|
static XGBoostRegressionTrainer.RegressionType |
XGBoostRegressionTrainer.RegressionType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static XGBoostRegressionTrainer.RegressionType[] |
XGBoostRegressionTrainer.RegressionType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Constructor and Description |
---|
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.
|
XGBoostRegressionTrainer(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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.