Struct slow_nn::Network

pub struct Network { /* fields omitted */ }

Neural Network struct

Methods

impl Network

pub fn from_conns(
    bias: f64,
    inputs: usize,
    outputs: usize,
    hidden: usize,
    conns: &[Connection]
) -> Self

Creates a network from a slice of Connections

pub fn dense(bias: f64, inputs: usize, outputs: usize, layers: &[usize]) -> Self

Creates a fully connected network with random weights

pub fn predict(&self, input: &[f64], activation: fn(_: f64) -> f64) -> Vec<f64>

Does a forward pass

pub fn train(
    &mut self,
    input: &[f64],
    expected: &[f64],
    activation: fn(_: f64) -> f64,
    deactivation: fn(_: f64) -> f64,
    loss: fn(_: f64, _: f64) -> f64,
    dloss: fn(_: f64, _: f64) -> f64,
    lr: f64
) -> f64

Does both forward and backward pass and updates the weights

pub fn load(path: &str) -> Result<Self, Error>

loads the network from a file

pub fn save(&self, path: &str) -> Result<(), Error>

Saves the network to a file

pub fn to_bytes(&self) -> Option<Vec<u8>>

Returns the bytes of the object after serializing

Trait Implementations

impl<'de> Deserialize<'de> for Network

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.

impl Serialize for Network

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

Serialize this value into the given Serde serializer.