org.tribuo.interop.tensorflow

Class RegressorTransformer

java.lang.Object

org.tribuo.interop.tensorflow.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 a Regressor to a Tensor containing a 32-bit float and a Tensor 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()
org.tensorflow.Tensor<?>	transform(List<Example<Regressor>> examples, ImmutableOutputInfo<Regressor> outputIDInfo) Converts a list of Example into a Tensor representing all the outputs in the list.
org.tensorflow.Tensor<?>	transform(Regressor example, ImmutableOutputInfo<Regressor> outputIDInfo) Converts an Output into a Tensor representing it's output.
List<Regressor>	transformToBatchOutput(org.tensorflow.Tensor<?> tensor, ImmutableOutputInfo<Regressor> outputIDInfo) Converts a Tensor containing multiple outputs into a list of Outputs.
List<Prediction<Regressor>>	transformToBatchPrediction(org.tensorflow.Tensor<?> tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int[] numValidFeatures, List<Example<Regressor>> examples) Converts a Tensor containing multiple outputs into a list of Predictions.
Regressor	transformToOutput(org.tensorflow.Tensor<?> tensor, ImmutableOutputInfo<Regressor> outputIDInfo) Converts a Tensor into the specified output type.
Prediction<Regressor>	transformToPrediction(org.tensorflow.Tensor<?> tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int numValidFeatures, Example<Regressor> example) Converts a Tensor 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(org.tensorflow.Tensor<?> tensor,
                                                   ImmutableOutputInfo<Regressor> outputIDInfo,
                                                   int numValidFeatures,
                                                   Example<Regressor> example)

Description copied from interface: OutputTransformer

Converts a Tensor into a Prediction.

Specified by:

transformToPrediction in interface OutputTransformer<Regressor>

Parameters:

tensor - The tensor 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(org.tensorflow.Tensor<?> tensor,
                                   ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

Converts a Tensor into the specified output type.

Specified by:

transformToOutput in interface OutputTransformer<Regressor>

Parameters:

tensor - The tensor to convert.

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

Returns:

A output.

transformToBatchPrediction

public List<Prediction<Regressor>> transformToBatchPrediction(org.tensorflow.Tensor<?> tensor,
                                                              ImmutableOutputInfo<Regressor> outputIDInfo,
                                                              int[] numValidFeatures,
                                                              List<Example<Regressor>> examples)

Description copied from interface: OutputTransformer

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

Specified by:

transformToBatchPrediction in interface OutputTransformer<Regressor>

Parameters:

tensor - The tensor 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(org.tensorflow.Tensor<?> tensor,
                                              ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

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

Specified by:

transformToBatchOutput in interface OutputTransformer<Regressor>

Parameters:

tensor - The tensor to convert.

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

Returns:

A list of outputs.

transform

public org.tensorflow.Tensor<?> transform(Regressor example,
                                          ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

Converts an Output into a Tensor representing it's output.

Specified by:

transform in interface OutputTransformer<Regressor>

Parameters:

example - The output to convert.

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

Returns:

A Tensor representing this output.

transform

public org.tensorflow.Tensor<?> transform(List<Example<Regressor>> examples,
                                          ImmutableOutputInfo<Regressor> outputIDInfo)

Description copied from interface: OutputTransformer

Converts a list of Example into a Tensor representing all the outputs in the list. It accepts a list of Example rather than a list of Output for efficiency reasons.

Specified by:

transform in interface OutputTransformer<Regressor>

Parameters:

examples - The examples to convert.

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

Returns:

A Tensor representing all the supplied 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>