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,
Map<String,com.oracle.labs.mlrg.olcut.provenance.Provenance> instanceProvenance)
Creates an
XGBoostExternalModel from the supplied in-memory XGBoost Booster . |
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)
Deprecated.
As the URL argument must always be valid. To wrap an in-memory booster use
createXGBoostModel(OutputFactory, Map, Map, XGBoostOutputConverter, Booster, Map) . |
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.