1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
pub mod layer; pub mod relu; pub mod tanh; pub mod dense; pub mod softmax; pub mod sigmoid; pub mod dropout; pub enum LayerEnum { Dense {input_units: usize, output_units: usize}, ReLU, Softmax, Sigmoid, TanH, Dropout {prob: f64} }