Package org.tribuo.interop.tensorflow

Class RegressorConverter

java.lang.Object
org.tribuo.interop.tensorflow.RegressorConverter
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, OutputConverter<Regressor>
public class RegressorConverter extends Object implements OutputConverter<Regressor>
Can convert a Regressor to a TFloat32 vector and a TFloat32 into a Prediction or Regressor.
See Also:

  • Constructor Details

    • RegressorConverter

      public RegressorConverter()
      Constructs a RegressorConverter.

  • Method Details

    • loss

      public BiFunction<org.tensorflow.op.Ops,com.oracle.labs.mlrg.olcut.util.Pair<org.tensorflow.op.core.Placeholder<? extends org.tensorflow.types.family.TNumber>,org.tensorflow.Operand<org.tensorflow.types.family.TNumber>>,org.tensorflow.Operand<org.tensorflow.types.family.TNumber>> loss()
      Returns a mean squared error loss.
      Specified by:
      loss in interface OutputConverter<Regressor>
      Returns:
      The mse loss.

    • outputTransformFunction

      public <U extends org.tensorflow.types.family.TNumber> BiFunction<org.tensorflow.op.Ops,org.tensorflow.Operand<U>,org.tensorflow.op.Op> outputTransformFunction()
      Applies the identity function
      Specified by:
      outputTransformFunction in interface OutputConverter<Regressor>
      Type Parameters:
      U - The input type (should be TFloat32).
      Returns:
      A function which applies the identity function.

    • convertToPrediction

      public Prediction<Regressor> convertToPrediction(org.tensorflow.Tensor tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int numValidFeatures, Example<Regressor> example)
      Description copied from interface: OutputConverter
      Converts a Tensor into a Prediction.
      Specified by:
      convertToPrediction in interface OutputConverter<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.

    • convertToOutput

      public Regressor convertToOutput(org.tensorflow.Tensor tensor, ImmutableOutputInfo<Regressor> outputIDInfo)
      Description copied from interface: OutputConverter
      Converts a Tensor into the specified output type.
      Specified by:
      convertToOutput in interface OutputConverter<Regressor>
      Parameters:
      tensor - The tensor to convert.
      outputIDInfo - The output info to use to identify the outputs.
      Returns:
      A output.

    • convertToBatchPrediction

      public List<Prediction<Regressor>> convertToBatchPrediction(org.tensorflow.Tensor tensor, ImmutableOutputInfo<Regressor> outputIDInfo, int[] numValidFeatures, List<Example<Regressor>> examples)
      Description copied from interface: OutputConverter
      Converts a Tensor containing multiple outputs into a list of Predictions.
      Specified by:
      convertToBatchPrediction in interface OutputConverter<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.

    • convertToBatchOutput

      public List<Regressor> convertToBatchOutput(org.tensorflow.Tensor tensor, ImmutableOutputInfo<Regressor> outputIDInfo)
      Description copied from interface: OutputConverter
      Converts a Tensor containing multiple outputs into a list of Outputs.
      Specified by:
      convertToBatchOutput in interface OutputConverter<Regressor>
      Parameters:
      tensor - The tensor to convert.
      outputIDInfo - The output info to use to identify the outputs.
      Returns:
      A list of outputs.

    • convertToTensor

      public org.tensorflow.Tensor convertToTensor(Regressor example, ImmutableOutputInfo<Regressor> outputIDInfo)
      Description copied from interface: OutputConverter
      Converts an Output into a Tensor representing it's output.
      Specified by:
      convertToTensor in interface OutputConverter<Regressor>
      Parameters:
      example - The output to convert.
      outputIDInfo - The output info to use to identify the outputs.
      Returns:
      A Tensor representing this output.

    • convertToTensor

      public org.tensorflow.Tensor convertToTensor(List<Example<Regressor>> examples, ImmutableOutputInfo<Regressor> outputIDInfo)
      Description copied from interface: OutputConverter
      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:
      convertToTensor in interface OutputConverter<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: OutputConverter
      Does this OutputConverter generate probabilities.
      Specified by:
      generatesProbabilities in interface OutputConverter<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>