[][src]Struct rail::model::ModelBuilder

pub struct ModelBuilder { /* fields omitted */ }

Methods

impl ModelBuilder[src]

pub fn layer(self, layer: Box<dyn Layer>) -> Self[src]

pub fn loss(self, loss_fn: Box<LossFuntionGenerator>) -> Self[src]

pub fn learning_rate(self, lr: Weight) -> Self[src]

pub fn input_size(self, amount: u64) -> Self[src]

pub fn input_shape(self, shape: &[u64; 4]) -> Self[src]

pub fn output_shape(self, shape: &[u64; 4]) -> Self[src]

pub fn build(self, init: bool) -> Result<Model, ModelError>[src]

Auto Trait Implementations

impl !Send for ModelBuilder

impl !Sync for ModelBuilder

Blanket Implementations

impl<T, U> Into<U> for T where
    U: From<T>, 
[src]

impl<T> From<T> for T[src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<T> BorrowMut<T> for T where
    T: ?Sized
[src]

impl<T> Borrow<T> for T where
    T: ?Sized
[src]

impl<T> Any for T where
    T: 'static + ?Sized
[src]