Uses of Class
org.tribuo.common.xgboost.XGBoostModel
Package
Description
Provides an interface to XGBoost for classification problems.
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts.Provides an interface to XGBoost for regression problems.
-
Uses of XGBoostModel in org.tribuo.classification.xgboost
Modifier and TypeMethodDescriptionXGBoostClassificationTrainer.train
(Dataset<Label> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) XGBoostClassificationTrainer.train
(Dataset<Label> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance, int invocationCount) -
Uses of XGBoostModel in org.tribuo.common.xgboost
Modifier and TypeMethodDescriptionprotected XGBoostModel<T>
XGBoostTrainer.createModel
(String name, ModelProvenance provenance, ImmutableFeatureMap featureIDMap, ImmutableOutputInfo<T> outputIDInfo, List<ml.dmlc.xgboost4j.java.Booster> models, XGBoostOutputConverter<T> converter) Creates an XGBoost model from the booster list.static XGBoostModel<?>
XGBoostModel.deserializeFromProto
(int version, String className, com.google.protobuf.Any message) Deserialization factory. -
Uses of XGBoostModel in org.tribuo.regression.xgboost
Modifier and TypeMethodDescriptionXGBoostRegressionTrainer.train
(Dataset<Regressor> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) XGBoostRegressionTrainer.train
(Dataset<Regressor> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance, int invocationCount)