Skip to main content

GatingNetwork

aprender::ensemble

Trait GatingNetwork 

Source 
pub trait GatingNetwork: Send + Sync {
    // Required methods
    fn forward(&self, x: &[f32]) -> Vec<f32>;
    fn n_features(&self) -> usize;
    fn n_experts(&self) -> usize;
}
Expand description

Trait for gating networks that route inputs to experts

Required Methods§

Source

fn forward(&self, x: &[f32]) -> Vec<f32>

Compute expert weights for input

Source

fn n_features(&self) -> usize

Number of input features

Source

fn n_experts(&self) -> usize

Number of experts

Implementors§