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 | Class and Description |
---|---|
class |
XGBoostClassificationConverter
Converts XGBoost outputs into
Label Prediction s. |
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) |
static <T extends Output<T>> |
XGBoostExternalModel.createXGBoostModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
XGBoostOutputConverter<T> outputFunc,
ml.dmlc.xgboost4j.java.Booster model,
URL provenanceLocation)
Creates an
XGBoostExternalModel from the supplied model. |
static <T extends Output<T>> |
XGBoostExternalModel.createXGBoostModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
XGBoostOutputConverter<T> outputFunc,
Path path)
Creates an
XGBoostExternalModel from the supplied model on disk. |
static <T extends Output<T>> |
XGBoostExternalModel.createXGBoostModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
XGBoostOutputConverter<T> outputFunc,
String path)
Creates an
XGBoostExternalModel from the supplied model on disk. |
Modifier and Type | Class and Description |
---|---|
class |
XGBoostRegressionConverter
Converts XGBoost outputs into
Regressor Prediction s. |
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.