Struct FeedForwardConfig 

pub struct FeedForwardConfig {
    pub d_model: usize,
    pub d_ff: usize,
    pub activation: String,
    pub dropout: f64,
}

Configuration for feed-forward network

Fields

d_model: usize

Model dimension (d_model)

d_ff: usize

Hidden dimension (typically 4 * d_model)

activation: String

Activation function name

dropout: f64

Dropout probability

Implementations

impl FeedForwardConfig

pub fn new(d_model: usize, d_ff: usize) -> Self

Create a new feed-forward configuration

pub fn with_activation(self, activation: impl Into<String>) -> Self

Set activation function

pub fn with_dropout(self, dropout: f64) -> Self

Set dropout probability

pub fn validate(&self) -> Result<()>

Validate configuration

Trait Implementations

impl Clone for FeedForwardConfig

fn clone(&self) -> FeedForwardConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FeedForwardConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for FeedForwardConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for FeedForwardConfig

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for FeedForwardConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for FeedForwardConfig