Package 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	Description
RegressorTransformer()	Constructs a RegressorTransformer.

Method Summary

Modifier and Type	Method	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 Details

RegressorTransformer

public RegressorTransformer()

Constructs a RegressorTransformer.

Method Details

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>