Struct Classification 

pub struct Classification<A> { /* private fields */ }

wrapper for converting loss to classification output

Implementations

impl<A> Classification<A>

pub fn from_inner(inner: A) -> Self

where Classification<A>: Op,

creates from the inner value

pub fn inner(&self) -> &A

references the inner value

pub fn inner_mut(&mut self) -> &mut A

references the inner value

pub fn into_inner(self) -> A

converts into the inner value

pub fn with_inner<B>(&self, inner: B) -> Classification<B>

where Classification<B>: Op,

replaces the inner value

Trait Implementations

impl<A: AI<X, LossOutput<B>>, B: Backend, X> AI<X, ClassificationOutput<B>> for Classification<A>

fn forward(&self, input: X) -> ClassificationOutput<B>

applies to the input

fn forward_mut(&mut self, input: X) -> ClassificationOutput<B>

applies to the input, possibly updating internal caches

impl<A: Clone> Clone for Classification<A>

fn clone(&self) -> Classification<A>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<A: Debug> Debug for Classification<A>

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

Formats the value using the given formatter.

impl<A: Decompose> Decompose for Classification<A>

type Decomposition = <A as Decompose>::Decomposition

the decomposed type

fn compose(decomposition: Self::Decomposition) -> Self

recreates from the decomposition

fn decompose(self) -> Self::Decomposition

owned decomposition

fn decompose_cloned(&self) -> Self::Decomposition

decomposition that copies data

impl<A: Default> Default for Classification<A>

fn default() -> Classification<A>

Returns the “default value” for a type.

impl<'de, A> Deserialize<'de> for Classification<A>

where A: Deserialize<'de>,

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<A: Op<Output = Y> + Wrappable, Y> Op for Classification<A>

where ClassificationLayer: AI<Y, ClassificationOutput<A::B>>,

type Output = ClassificationOutput<<A as Wrappable>::B>

suggested output type to help with composition coherence. Ideally, Self should implement AI<X,Self::Output> for some X

fn chain<B>(self, b: B) -> Sequential<(Self, B)>

where Self: Sized, Sequential<(Self, B)>: Op,

sequences with another ai operation

fn duplicate(self) -> Duplicate<Self>

where Duplicate<Self>: Op, Self: Sized,

wraps with a duplicate operation

fn fix_type<Z>(self) -> SetType<Self, Z, Z>

where Self: AI<Z, Z> + Sized,

set type but with the same input and output

fn flatten<R: Clone>(self, args: R) -> Flatten<Self, R>

where Flatten<Self, R>: Op, Self: Sized,

wraps with a flatten layer

fn forward_fixed<Z>(&self, input: Z) -> Z

where Self: AI<Z, Z> + Sized,

applies to the input

fn forward_fixed_mut<Z>(&mut self, input: Z) -> Z

where Self: AI<Z, Z> + Sized,

applies to the input

fn forward_typed<W, Z>(&self, input: W) -> Z

where Self: AI<W, Z> + Sized,

applies to the input

fn forward_typed_mut<W, Z>(&mut self, input: W) -> Z

where Self: AI<W, Z> + Sized,

applies to the input, possibly updating internal caches

fn infer_autoregressive<X, Y>(self, input: X) -> Autoregression<Self, Y>

where Self: AI<X, Y> + AI<Y, Y> + Sized, Y: Clone,

creates an autoregressive inference

fn map<B>(self, b: B) -> Map<Sequential<(Self, B)>>

where Map<Sequential<(Self, B)>>: Op, Self: Sized, Sequential<(Self, B)>: Op,

applies the operation to every output

fn optional(self) -> Option<Self>

where Self: Sized,

creates an optional operation

fn reshape<R: Clone>(self, args: R) -> Reshape<Self, R>

where Reshape<Self, R>: Op, Self: Sized,

wraps with a reshape layer

fn residual(self) -> Residual<Self>

where Residual<Self>: Op, Self: Sized,

wraps in a residual layer

fn set_type<W, Z>(self) -> SetType<Self, W, Z>

where Self: AI<W, Z> + Sized,

sets the input output types

fn to_each(self) -> Map<Self>

where Map<Self>: Op, Self: Sized,

wraps with a map operation

fn wrap_inner(self) -> Inner<Self>

where Self: Sized,

wraps the inner value so it can be unwrapped with unwrap inner

fn zip<B>(self, b: B) -> Zip<(Self, B)>

where Self: Sized, Zip<(Self, B)>: Op,

zips with another ai operation

impl<A> Serialize for Classification<A>

where A: Serialize,

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl<A: UnwrapInner> UnwrapInner for Classification<A>

type Inner = <A as UnwrapInner>::Inner

the inner type

fn unwrap_inner(self) -> Self::Inner

unwraps the inner value

impl<W: Wrappable> Wrappable for Classification<W>

type B = <W as Wrappable>::B

type With<C: Backend> = Classification<<W as Wrappable>::With<C>>

impl<A: Copy> Copy for Classification<A>