Trait ModelExt

pub trait ModelExt<T>: Model<T>
where
    Self::Layout: ModelLayout,
{
    // Provided methods
    fn replace_config(&mut self, config: Self::Config) -> Self::Config { ... }
    fn replace_params(
        &mut self,
        params: DeepModelParams<T>,
    ) -> DeepModelParams<T> { ... }
    fn set_config(&mut self, config: Self::Config) -> &mut Self { ... }
    fn set_params(&mut self, params: DeepModelParams<T>) -> &mut Self { ... }
    fn input_layer(&self) -> &Params<T> { ... }
    fn input_layer_mut(&mut self) -> &mut Params<T> { ... }
    fn hidden_layers(&self) -> &Vec<Params<T>>  { ... }
    fn hidden_layers_mut(&mut self) -> &mut Vec<Params<T>>  { ... }
    fn output_layer(&self) -> &Params<T> { ... }
    fn output_layer_mut(&mut self) -> &mut Params<T> { ... }
    fn set_input_layer(&mut self, layer: Params<T>) -> &mut Self { ... }
    fn set_hidden_layers(&mut self, layers: Vec<Params<T>>) -> &mut Self { ... }
    fn set_output_layer(&mut self, layer: Params<T>) -> &mut Self { ... }
    fn input_dim(&self) -> (usize, usize) { ... }
    fn hidden_dim(&self) -> (usize, usize) { ... }
    fn hidden_layers_count(&self) -> usize { ... }
    fn output_dim(&self) -> (usize, usize) { ... }
}

Provided Methods

fn replace_config(&mut self, config: Self::Config) -> Self::Config

replace the current configuration and returns the old one;

fn replace_params(&mut self, params: DeepModelParams<T>) -> DeepModelParams<T>

replace the current model parameters and returns the old one

fn set_config(&mut self, config: Self::Config) -> &mut Self

overrides the current configuration and returns a mutable reference to the model

fn set_params(&mut self, params: DeepModelParams<T>) -> &mut Self

overrides the current model parameters and returns a mutable reference to the model

fn input_layer(&self) -> &Params<T>

returns an immutable reference to the input layer;

fn input_layer_mut(&mut self) -> &mut Params<T>

returns a mutable reference to the input layer;

fn hidden_layers(&self) -> &Vec<Params<T>>

returns an immutable reference to the hidden layer(s);

fn hidden_layers_mut(&mut self) -> &mut Vec<Params<T>>

returns a mutable reference to the hidden layer(s);

fn output_layer(&self) -> &Params<T>

returns an immutable reference to the output layer;

fn output_layer_mut(&mut self) -> &mut Params<T>

returns a mutable reference to the output layer;

fn set_input_layer(&mut self, layer: Params<T>) -> &mut Self

fn set_hidden_layers(&mut self, layers: Vec<Params<T>>) -> &mut Self

fn set_output_layer(&mut self, layer: Params<T>) -> &mut Self

fn input_dim(&self) -> (usize, usize)

returns a 2-tuple representing the dimensions of the input layer; (input, hidden)

fn hidden_dim(&self) -> (usize, usize)

returns a 2-tuple representing the dimensions of the hidden layers; (hidden, hidden)

fn hidden_layers_count(&self) -> usize

returns the total number of hidden layers in the model;

fn output_dim(&self) -> (usize, usize)

returns a 2-tuple representing the dimensions of the output layer; (hidden, output)

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<M, T> ModelExt<T> for M

where M: Model<T>, M::Layout: ModelLayout,