org.tribuo.interop.onnx

Class RegressorTransformer

java.lang.Object

org.tribuo.interop.onnx.RegressorTransformer

All Implemented Interfaces:

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

public class RegressorTransformer
extends Object
implements OutputTransformer<Regressor>

Can convert an OnnxValue into a Prediction or Regressor.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
RegressorTransformer()

Method Summary

Modifier and Type	Method and Description
boolean	generatesProbabilities() Does this OutputTransformer generate probabilities.
com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance	getProvenance()
String	toString()
List<Regressor>	transformToBatchOutput(List<ai.onnxruntime.OnnxValue> tensor, ImmutableOutputInfo<Regressor> outputIDInfo) Converts a OnnxValue containing multiple outputs into a list of Outputs.
List<Prediction<Regressor>>	transformToBatchPrediction(List<ai.onnxruntime.OnnxValue> tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int[] numValidFeatures, List<Example<Regressor>> examples) Converts a OnnxValue containing multiple outputs into a list of Predictions.
Regressor	transformToOutput(List<ai.onnxruntime.OnnxValue> tensor, ImmutableOutputInfo<Regressor> outputIDInfo) Converts a OnnxValue into the specified output type.
Prediction<Regressor>	transformToPrediction(List<ai.onnxruntime.OnnxValue> tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int numValidFeatures, Example<Regressor> example) Converts a OnnxValue into a Prediction.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

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

postConfig

Constructor Detail

RegressorTransformer

public RegressorTransformer()

Method Detail

transformToPrediction

public Prediction<Regressor> transformToPrediction(List<ai.onnxruntime.OnnxValue> tensor,
                                                   ImmutableOutputInfo<Regressor> outputIDInfo,
                                                   int numValidFeatures,
                                                   Example<Regressor> example)

Description copied from interface: OutputTransformer

Converts a OnnxValue into a Prediction.

Specified by:

transformToPrediction in interface OutputTransformer<Regressor>

Parameters:

tensor - 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

public Regressor transformToOutput(List<ai.onnxruntime.OnnxValue> tensor,
                                   ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

Converts a OnnxValue into the specified output type.

Specified by:

transformToOutput in interface OutputTransformer<Regressor>

Parameters:

tensor - The value to convert.

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

Returns:

A output.

transformToBatchPrediction

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

Description copied from interface: OutputTransformer

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

Specified by:

transformToBatchPrediction in interface OutputTransformer<Regressor>

Parameters:

tensor - 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

public List<Regressor> transformToBatchOutput(List<ai.onnxruntime.OnnxValue> tensor,
                                              ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

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

Specified by:

transformToBatchOutput in interface OutputTransformer<Regressor>

Parameters:

tensor - The value to convert.

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

Returns:

A list of outputs.

generatesProbabilities

public boolean generatesProbabilities()

Description copied from interface: OutputTransformer

Does this OutputTransformer generate probabilities.

Specified by:

generatesProbabilities in interface OutputTransformer<Regressor>

Returns:

True if it produces a probability distribution in the Prediction.

toString

public String toString()

Overrides:

toString in class Object

getProvenance

public com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance getProvenance()

Specified by:

getProvenance in interface com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>