Struct MLPConfig 

pub struct MLPConfig {
    pub input_dim: usize,
    pub output_dim: usize,
    pub hidden_dims: Vec<usize>,
    pub activation: Activation,
    pub output_activation: Option<Activation>,
}

Configuration for building an MLP.

Fields

input_dim: usize

output_dim: usize

hidden_dims: Vec<usize>

activation: Activation

output_activation: Option<Activation>

Implementations

impl MLPConfig

pub fn new(input_dim: usize, output_dim: usize) -> Self

pub fn with_hidden(self, dims: Vec<usize>) -> Self

pub fn with_activation(self, act: Activation) -> Self

pub fn with_output_activation(self, act: Activation) -> Self

Trait Implementations

impl Clone for MLPConfig

fn clone(&self) -> MLPConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MLPConfig

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

Formats the value using the given formatter.