Struct LayerConfig 

pub struct LayerConfig {
    pub layer_type: LayerType,
    pub units: Option<usize>,
    pub kernel_size: Option<usize>,
    pub activation: Option<String>,
    pub dropout_rate: Option<f64>,
    pub active: bool,
}

Configuration for a single layer in the architecture.

Fields

layer_type: LayerType

Type of layer

units: Option<usize>

Number of units/filters (for Dense/Conv2d)

kernel_size: Option<usize>

Kernel size (for Conv2d)

activation: Option<String>

Activation function name

dropout_rate: Option<f64>

Dropout rate (for Dropout layer)

active: bool

Whether this layer is active (for pruning)

Implementations

impl LayerConfig

pub fn new(layer_type: LayerType) -> LayerConfig

Create a new layer configuration.

pub fn with_units(self, units: usize) -> LayerConfig

Set number of units/filters.

pub fn with_kernel_size(self, size: usize) -> LayerConfig

Set kernel size.

pub fn with_activation(self, activation: &str) -> LayerConfig

Set activation function.

pub fn with_dropout_rate(self, rate: f64) -> LayerConfig

Set dropout rate.

Trait Implementations

impl Clone for LayerConfig

fn clone(&self) -> LayerConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LayerConfig

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

Formats the value using the given formatter.

impl PartialEq for LayerConfig

fn eq(&self, other: &LayerConfig) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for LayerConfig