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:
LabelOneVOneTransformer,LabelTransformer,MultiLabelTransformer,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 TypeMethodDescriptionbooleanDoes this OutputTransformer generate probabilities.transformToBatchOutput(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo) Converts aOnnxValuecontaining multiple outputs into a list ofOutputs.List<Prediction<T>> transformToBatchPrediction(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo, int[] numValidFeatures, List<Example<T>> examples) Converts aOnnxValuecontaining multiple outputs into a list ofPredictions.transformToOutput(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo) Converts aOnnxValueinto the specified output type.transformToPrediction(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo, int numValidFeatures, Example<T> example) Converts aOnnxValueinto aPrediction.Methods inherited from interface com.oracle.labs.mlrg.olcut.config.Configurable
postConfigMethods inherited from interface com.oracle.labs.mlrg.olcut.provenance.Provenancable
getProvenance
-
Method Details
-
transformToPrediction
Prediction<T> transformToPrediction(List<ai.onnxruntime.OnnxValue> value, ImmutableOutputInfo<T> outputIDInfo, int numValidFeatures, Example<T> example) Converts aOnnxValueinto aPrediction.- 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
Converts aOnnxValueinto 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 aOnnxValuecontaining multiple outputs into a list ofPredictions.- 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 aOnnxValuecontaining multiple outputs into a list ofOutputs.- 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.
-