kratom 1.0.1

Simple neural network library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

pub mod standard;



pub enum NeuralNetworkError {
    BadBufferSize,
}


pub trait NeuralNetwork {
    fn forward(&mut self, input: &[f32]) -> Vec<f32>;
    fn backward(&mut self, input: &[f32], expected: &[f32]) -> (Vec<f32>, Vec<f32>, Vec<f32>, Vec<f32>, Vec<f32>, Vec<f32>, f32);
    fn train(&mut self, batch: &[(Vec<f32>, Vec<f32>)], eta: f32) -> Vec<f32>;

    fn forward_buf(&mut self, input: &[f32], output_buf: &mut [f32]) -> Result<(), NeuralNetworkError>;
    //fn backward_buf(&mut self, input: &[f32], expected: &[f32], grad_bufs: &mut [Vec<f32>], delta_bufs: &mut [Vec<f32>]) -> f32;
    //fn train_buf(&mut self, batch: &[(Vec<f32>, Vec<f32>)], eta: f32, errors_buf: &mut [f32]);
}