rail 0.3.1

A GPU-Accelerated ML Library akin to Keras
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

pub mod dense;
pub mod conv;
pub mod activations;

use crate::{ Weight, Matrix };
use crate::layers::activations::Activation;


// TODO: refactor to const strings from each layer
pub fn load_layer(src: &[&str]) -> Box<Layer> {
    use crate::layers::dense::Dense;

    match src[0] {
        "dense" => Dense::deserialize(&src[1..]),
        v => panic!("wrong kind of layer {:?}!", v),
    }

}

pub trait Layer {
    fn feedforward(&self, input: Matrix) -> Matrix;
    fn backpropagate(&mut self, input: &Matrix, output: &Matrix, d_error: Matrix, lr: Weight) -> Matrix;
    fn initialize(&mut self, inputs: &[u64; 4]);
    fn output_shape(&self) -> [u64; 4];
    fn serialize(&self) -> String;

    fn a_function(&self) -> Activation;

    fn display(&self);
}