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 | Method and Description |
---|---|
XGBoostModel<Label> |
XGBoostClassificationTrainer.train(Dataset<Label> examples,
Map<String,com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) |
Modifier and Type | Method and Description |
---|---|
protected XGBoostModel<T> |
XGBoostTrainer.createModel(String name,
ModelProvenance provenance,
ImmutableFeatureMap featureIDMap,
ImmutableOutputInfo<T> outputIDInfo,
List<ml.dmlc.xgboost4j.java.Booster> models,
XGBoostOutputConverter<T> converter) |
Modifier and Type | Method and Description |
---|---|
XGBoostModel<Regressor> |
XGBoostRegressionTrainer.train(Dataset<Regressor> examples,
Map<String,com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) |
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.