Package | Description |
---|---|
org.tribuo.interop.onnx |
This package contains a Tribuo wrapper around the ONNX Runtime.
|
Modifier and Type | Class and Description |
---|---|
class |
DenseTransformer
Converts a sparse Tribuo example into a dense float vector, then wraps it in an
OnnxTensor . |
class |
ImageTransformer
Image transformer.
|
Modifier and Type | Method and Description |
---|---|
static <T extends Output<T>> |
ONNXExternalModel.createOnnxModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
ExampleTransformer featureTransformer,
OutputTransformer<T> outputTransformer,
ai.onnxruntime.OrtSession.SessionOptions opts,
Path path,
String inputName)
Creates an
ONNXExternalModel by loading the model from disk. |
static <T extends Output<T>> |
ONNXExternalModel.createOnnxModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
ExampleTransformer featureTransformer,
OutputTransformer<T> outputTransformer,
ai.onnxruntime.OrtSession.SessionOptions opts,
String filename,
String inputName)
Creates an
ONNXExternalModel by loading the model from disk. |
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.