Tinkex.Types.ForwardBackwardOutput (Tinkex v0.3.4)

View Source

Output from forward-backward pass.

Mirrors Python tinker.types.ForwardBackwardOutput.

NOTE: There is NO loss field. Loss is accessed via metrics["loss"].

Summary

Types

t()

Functions

from_json(json)

Parse a forward-backward output from JSON.

loss(forward_backward_output)

Get the loss value from metrics.

Types

t()

@type t() :: %Tinkex.Types.ForwardBackwardOutput{
  loss_fn_output_type: String.t(),
  loss_fn_outputs: [map()],
  metrics: %{required(String.t()) => float()}
}

Functions

from_json(json)

@spec from_json(map()) :: t()

Parse a forward-backward output from JSON.

loss(forward_backward_output)

@spec loss(t()) :: float() | nil

Get the loss value from metrics.