Index
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form
C
- convertBatchOutput(ImmutableOutputInfo<Label>, List<float[][]>, int[], Example<Label>[]) - Method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
- convertOutput(ImmutableOutputInfo<Label>, List<float[]>, int, Example<Label>) - Method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
- CURRENT_VERSION - Static variable in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
-
Protobuf serialization version.
D
- deserializeFromProto(int, String, Any) - Static method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
-
Deserialization factory.
G
- general - Variable in class org.tribuo.classification.xgboost.TrainTest.TrainTestOptions
-
The data loading options.
- generatesProbabilities() - Method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
- getOptionsDescription() - Method in class org.tribuo.classification.xgboost.TrainTest.TrainTestOptions
- getProvenance() - Method in class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
- getTrainer() - Method in class org.tribuo.classification.xgboost.XGBoostOptions
- getTypeWitness() - Method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
M
- main(String[]) - Static method in class org.tribuo.classification.xgboost.TrainTest
-
Runs a TrainTest CLI.
O
- org.tribuo.classification.xgboost - package org.tribuo.classification.xgboost
-
Provides an interface to XGBoost for classification problems.
P
- postConfig() - Method in class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
Used by the OLCUT configuration system, and should not be called by external code.
S
- serialize() - Method in class org.tribuo.classification.xgboost.XGBoostClassificationConverter
T
- train(Dataset<Label>) - Method in class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
- train(Dataset<Label>, Map<String, Provenance>) - Method in class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
- train(Dataset<Label>, Map<String, Provenance>, int) - Method in class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
- TrainTest - Class in org.tribuo.classification.xgboost
-
Build and run an XGBoost classifier for a standard dataset.
- TrainTest() - Constructor for class org.tribuo.classification.xgboost.TrainTest
- TrainTest.TrainTestOptions - Class in org.tribuo.classification.xgboost
-
Command line options.
- TrainTestOptions() - Constructor for class org.tribuo.classification.xgboost.TrainTest.TrainTestOptions
X
- xbgAlpha - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
L1 regularization term for weights.
- xgbBoosterType - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Weak learning algorithm.
- xgbEnsembleSize - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Number of trees in the ensemble.
- xgbEta - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Step size shrinkage parameter (range [0,1]).
- xgbGamma - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Minimum loss reduction to make a split (range [0,Infinity]).
- xgbLambda - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
L2 regularization term for weights.
- xgbLogLevel - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Make the XGBoost training procedure quiet.
- xgbMaxDepth - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Max tree depth (range (0,Integer.MAX_VALUE]).
- xgbMinWeight - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Minimum sum of instance weights needed in a leaf (range [0,Infinity]).
- xgbNumThreads - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Number of threads to use (range (1, num hw threads)).
- XGBoostClassificationConverter - Class in org.tribuo.classification.xgboost
-
Converts XGBoost outputs into
LabelPredictions. - XGBoostClassificationConverter() - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationConverter
-
Constructs an XGBoostClassificationConverter.
- XGBoostClassificationTrainer - Class in org.tribuo.classification.xgboost
-
A
Trainerwhich wraps the XGBoost training procedure. - XGBoostClassificationTrainer() - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
For olcut.
- XGBoostClassificationTrainer(int) - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
Create an XGBoost trainer.
- XGBoostClassificationTrainer(int, double, double, int, double, double, double, double, double, int, boolean, long) - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
Create an XGBoost trainer.
- XGBoostClassificationTrainer(int, int, boolean) - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
Create an XGBoost trainer.
- XGBoostClassificationTrainer(int, Map<String, Object>) - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
This gives direct access to the XGBoost parameter map.
- XGBoostClassificationTrainer(XGBoostTrainer.BoosterType, XGBoostTrainer.TreeMethod, int, double, double, int, double, double, double, double, double, int, XGBoostTrainer.LoggingVerbosity, long) - Constructor for class org.tribuo.classification.xgboost.XGBoostClassificationTrainer
-
Create an XGBoost trainer.
- xgboostOptions - Variable in class org.tribuo.classification.xgboost.TrainTest.TrainTestOptions
-
The XGBoost trainer options.
- XGBoostOptions - Class in org.tribuo.classification.xgboost
-
CLI options for training an XGBoost classifier.
- XGBoostOptions() - Constructor for class org.tribuo.classification.xgboost.XGBoostOptions
- xgbQuiet - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Deprecated, use xgb-loglevel.
- xgbSubsample - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Subsample size for each tree (range (0,1]).
- xgbSubsampleFeatures - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Subsample features for each tree (range (0,1]).
- xgbTreeMethod - Variable in class org.tribuo.classification.xgboost.XGBoostOptions
-
Tree building algorithm.
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form