Struct MlpBuilder 

pub struct MlpBuilder { /* private fields */ }

Builder for an Mlp.

Example:

use rust_mlp::{Activation, MlpBuilder};

let mlp = MlpBuilder::new(2)?
    .add_layer(8, Activation::ReLU)?
    .add_layer(1, Activation::Sigmoid)?
    .build_with_seed(0)?;

Implementations

impl MlpBuilder

pub fn new(input_dim: usize) -> Result<Self>

Start building an MLP that accepts inputs of length input_dim.

pub fn from_sizes(sizes: &[usize], activations: &[Activation]) -> Result<Self>

Convenience constructor from a sizes list + activations.

sizes includes input and output dimensions, so its length must be at least 2. activations must have length sizes.len() - 1.

pub fn add_layer(self, out_dim: usize, activation: Activation) -> Result<Self>

Add a dense layer.

The layer will have out_dim outputs and uses activation.

pub fn build_with_seed(self, seed: u64) -> Result<Mlp>

Build using a deterministic seed.

pub fn build_with_rng<R: Rng + ?Sized>(self, rng: &mut R) -> Result<Mlp>

Build using the provided RNG.

Trait Implementations

impl Clone for MlpBuilder

fn clone(&self) -> MlpBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MlpBuilder

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

Formats the value using the given formatter.