Enum NetworkArchitecture 

pub enum NetworkArchitecture {
    MLP {
        hidden_layers: Vec<usize>,
    },
    CNN {
        conv_layers: Vec<ConvLayer>,
        fc_layers: Vec<usize>,
    },
    RNN {
        rnn_type: String,
        hidden_size: usize,
        num_layers: usize,
    },
    Transformer {
        num_heads: usize,
        hidden_dim: usize,
        num_layers: usize,
    },
    Autoencoder {
        encoder_layers: Vec<usize>,
        decoder_layers: Vec<usize>,
    },
}

Network architecture templates

Variants

MLP

Fields

hidden_layers: Vec<usize>

CNN

Fields

conv_layers: Vec<ConvLayer>

fc_layers: Vec<usize>

RNN

Fields

rnn_type: String

hidden_size: usize

num_layers: usize

Transformer

Fields

num_heads: usize

hidden_dim: usize

num_layers: usize

Autoencoder

Fields

encoder_layers: Vec<usize>

decoder_layers: Vec<usize>

Trait Implementations

impl Clone for NetworkArchitecture

fn clone(&self) -> NetworkArchitecture

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NetworkArchitecture

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

Formats the value using the given formatter.