Package | Description |
---|---|
org.tribuo.common.xgboost |
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts. |
Modifier and Type | Method and Description |
---|---|
protected XGBoostExternalModel<T> |
XGBoostExternalModel.copy(String newName,
ModelProvenance newProvenance) |
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. |
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.