Uses of Class
org.tribuo.common.xgboost.XGBoostModel
Packages that use 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
Methods in org.tribuo.classification.xgboost that return XGBoostModelModifier 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
Methods in org.tribuo.common.xgboost that return XGBoostModelModifier 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
Methods in org.tribuo.regression.xgboost that return XGBoostModelModifier 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)