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) |
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) |
XGBoostRegressionTrainer(XGBoostRegressionTrainer.RegressionType rType,
int numTrees,
Map<String,Object> parameters)
This gives direct access to the XGBoost parameter map.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.