Struct Model 

pub struct Model {
    pub layers: Vec<Layer>,
    pub name: String,
}

A sequential neural network model.

Fields

layers: Vec<Layer>

name: String

Implementations

impl Model

pub fn new(name: &str) -> Self

pub fn forward(&self, input: &Tensor) -> Tensor

pub fn parameter_count(&self) -> usize

pub fn save_weights(&self, path: &str) -> Result<(), String>

Save weights to a simple binary format: For each tensor: [ndim: u32] [shape[0]: u32] … [shape[n-1]: u32] [data as f32 LE bytes]

pub fn load_weights(&mut self, path: &str) -> Result<(), String>

Load weights from the binary format written by save_weights.

Trait Implementations

impl Clone for Model

fn clone(&self) -> Model

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Model

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

Formats the value using the given formatter.