Package org.tribuo.interop.onnx
package org.tribuo.interop.onnx
This package contains a Tribuo wrapper around the ONNX Runtime.
The ONNX Runtime allows the deployment of models trained in scikit-learn, pytorch and other packages without requiring them to be present at runtime.
N.B. ONNX support is experimental, and may change without a major version bump.
-
ClassDescriptionConverts a sparse Tribuo example into a dense float vector, then wraps it in an
OnnxTensor
.Transforms aSparseVector
, extracting the features from it as aOnnxTensor
.Image transformer.ONNXExternalModel<T extends Output<T>>A Tribuo wrapper around a ONNX model.OutputTransformer<T extends Output<T>>