Package | Description |
---|---|
org.tribuo.classification.xgboost |
Provides an interface to XGBoost for classification problems.
|
org.tribuo.common.xgboost |
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts. |
org.tribuo.regression.xgboost |
Provides an interface to XGBoost for regression problems.
|
Modifier and Type | Field and Description |
---|---|
XGBoostTrainer.TreeMethod |
XGBoostOptions.xgbTreeMethod |
Constructor and Description |
---|
XGBoostClassificationTrainer(XGBoostTrainer.BoosterType boosterType,
XGBoostTrainer.TreeMethod treeMethod,
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.
|
Modifier and Type | Method and Description |
---|---|
static XGBoostTrainer.TreeMethod |
XGBoostTrainer.TreeMethod.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static XGBoostTrainer.TreeMethod[] |
XGBoostTrainer.TreeMethod.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Constructor and Description |
---|
XGBoostTrainer(XGBoostTrainer.BoosterType boosterType,
XGBoostTrainer.TreeMethod treeMethod,
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.
|
Constructor and Description |
---|
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.