org.tribuo.interop.onnx

Interface OutputTransformer<T extends Output<T>>

All Superinterfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>, Serializable

All Known Implementing Classes:

LabelTransformer, RegressorTransformer

public interface OutputTransformer<T extends Output<T>>
extends com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>, Serializable

Converts an OnnxValue into an Output or a Prediction.

N.B. ONNX support is experimental, and may change without a major version bump.

Method Summary

Modifier and Type	Method and Description
boolean	generatesProbabilities() Does this OutputTransformer generate probabilities.
List<T>	transformToBatchOutput(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo) Converts a OnnxValue containing multiple outputs into a list of Outputs.
List<Prediction<T>>	transformToBatchPrediction(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo, int[] numValidFeatures, List<Example<T>> examples) Converts a OnnxValue containing multiple outputs into a list of Predictions.
T	transformToOutput(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo) Converts a OnnxValue into the specified output type.
Prediction<T>	transformToPrediction(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo, int numValidFeatures, Example<T> example) Converts a OnnxValue into a Prediction.

Methods inherited from interface com.oracle.labs.mlrg.olcut.config.Configurable

postConfig

Methods inherited from interface com.oracle.labs.mlrg.olcut.provenance.Provenancable

getProvenance

Method Detail

transformToPrediction

Prediction<T> transformToPrediction(List<ai.onnxruntime.OnnxValue> value,
                                    ImmutableOutputInfo<T> outputIDInfo,
                                    int numValidFeatures,
                                    Example<T> example)

Converts a OnnxValue into a Prediction.

Parameters:

value - The value to convert.

outputIDInfo - The output info to use to identify the outputs.

numValidFeatures - The number of valid features used by the prediction.

example - The example to insert into the prediction.

Returns:

A prediction object.

transformToOutput

T transformToOutput(List<ai.onnxruntime.OnnxValue> value,
                    ImmutableOutputInfo<T> outputIDInfo)

Converts a OnnxValue into the specified output type.

Parameters:

value - The value to convert.

outputIDInfo - The output info to use to identify the outputs.

Returns:

A output.

transformToBatchPrediction

List<Prediction<T>> transformToBatchPrediction(List<ai.onnxruntime.OnnxValue> value,
                                               ImmutableOutputInfo<T> outputIDInfo,
                                               int[] numValidFeatures,
                                               List<Example<T>> examples)

Converts a OnnxValue containing multiple outputs into a list of Predictions.

Parameters:

value - The value to convert.

outputIDInfo - The output info to use to identify the outputs.

numValidFeatures - The number of valid features used by the prediction.

examples - The example to insert into the prediction.

Returns:

A list of predictions.

transformToBatchOutput

List<T> transformToBatchOutput(List<ai.onnxruntime.OnnxValue> value,
                               ImmutableOutputInfo<T> outputIDInfo)

Converts a OnnxValue containing multiple outputs into a list of Outputs.

Parameters:

value - The value to convert.

outputIDInfo - The output info to use to identify the outputs.

Returns:

A list of outputs.

generatesProbabilities

boolean generatesProbabilities()

Does this OutputTransformer generate probabilities.

Returns:

True if it produces a probability distribution in the Prediction.