Interface ForwardConfig.Builder

  • Method Details

    • lossFnName

      ForwardConfig.Builder lossFnName(String lossFnName)

      The name of the loss function to evaluate during the forward pass.

      Parameters:
      lossFnName - The name of the loss function to evaluate during the forward pass.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
      See Also:
    • lossFnName

      ForwardConfig.Builder lossFnName(LossFnName lossFnName)

      The name of the loss function to evaluate during the forward pass.

      Parameters:
      lossFnName - The name of the loss function to evaluate during the forward pass.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
      See Also:
    • datumList

      ForwardConfig.Builder datumList(Collection<Datum> datumList)

      The list of input data to feed through the model. Must contain at least one datum, and each datum must have a non-empty modelInput with at least one chunk.

      Parameters:
      datumList - The list of input data to feed through the model. Must contain at least one datum, and each datum must have a non-empty modelInput with at least one chunk.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • datumList

      ForwardConfig.Builder datumList(Datum... datumList)

      The list of input data to feed through the model. Must contain at least one datum, and each datum must have a non-empty modelInput with at least one chunk.

      Parameters:
      datumList - The list of input data to feed through the model. Must contain at least one datum, and each datum must have a non-empty modelInput with at least one chunk.
      Returns:
      Returns a reference to this object so that method calls can be chained together.
    • datumList

      ForwardConfig.Builder datumList(Consumer<Datum.Builder>... datumList)

      The list of input data to feed through the model. Must contain at least one datum, and each datum must have a non-empty modelInput with at least one chunk.

      This is a convenience method that creates an instance of the Datum.Builder avoiding the need to create one manually via Datum.builder().

      When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to datumList(List<Datum>).

      Parameters:
      datumList - a consumer that will call methods on Datum.Builder
      Returns:
      Returns a reference to this object so that method calls can be chained together.
      See Also:
    • lossFnConfig

      ForwardConfig.Builder lossFnConfig(Document lossFnConfig)

      Additional loss function configuration passed to the engine. Supported keys are clip_low_threshold (valid range: 0 to 1) and clip_high_threshold (valid range: 1 to 10). Unsupported keys are rejected.

      Parameters:
      lossFnConfig - Additional loss function configuration passed to the engine. Supported keys are clip_low_threshold (valid range: 0 to 1) and clip_high_threshold (valid range: 1 to 10). Unsupported keys are rejected.
      Returns:
      Returns a reference to this object so that method calls can be chained together.