Struct BinaryCrossEntropyBuilder

pub struct BinaryCrossEntropyBuilder<'a> { /* private fields */ }

Builder for BinaryCrossEntropy.

Implementations

impl<'a> BinaryCrossEntropyBuilder<'a>

pub fn new() -> Self

Creates a new BinaryCrossEntropyBuilder.

impl<'a> BinaryCrossEntropyBuilder<'a>

pub fn weights(self, weights: impl Into<Option<&'a Array>>) -> Self

Sets the value of [weights].

pub fn inputs_are_logits(self, inputs_are_logits: impl Into<bool>) -> Self

Sets the value of [inputs_are_logits].

pub fn reduction(self, reduction: impl Into<LossReduction>) -> Self

Sets the value of [reduction].

Trait Implementations

impl<'a> Builder<BinaryCrossEntropy<'a>> for BinaryCrossEntropyBuilder<'a>

type Error = Infallible

Error with building

fn build(self) -> Result<BinaryCrossEntropy<'a>, Self::Error>

Build the type

impl<'a> Clone for BinaryCrossEntropyBuilder<'a>

fn clone(&self) -> BinaryCrossEntropyBuilder<'a>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<'a> Debug for BinaryCrossEntropyBuilder<'a>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.