Package | Description |
---|---|
org.tribuo.common.xgboost |
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts. |
org.tribuo.interop.onnx |
This package contains a Tribuo wrapper around the ONNX Runtime.
|
org.tribuo.interop.tensorflow |
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
|
Modifier and Type | Class and Description |
---|---|
class |
XGBoostExternalModel<T extends Output<T>>
A
Model which wraps around a XGBoost.Booster which was trained by a system other than Tribuo. |
Modifier and Type | Class and Description |
---|---|
class |
ONNXExternalModel<T extends Output<T>>
A Tribuo wrapper around a ONNX model.
|
Modifier and Type | Class and Description |
---|---|
class |
TensorFlowFrozenExternalModel<T extends Output<T>>
A Tribuo wrapper around a TensorFlow frozen model.
|
class |
TensorFlowSavedModelExternalModel<T extends Output<T>>
A Tribuo wrapper around a TensorFlow saved model bundle.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.