easynn 0.1.3-alpha

(Pre-alpha) A rust crate for quick-and-dirty neuro network developing and training.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

//! The models module
//! 

pub mod sequential;

pub mod losses;

use crate::layers::*;
pub use losses::*;

pub trait Model<T: NumT>  {
    /// Calculate the output according to the input
    fn predict(&self, input: &Tensor<T>) -> Result<Tensor<T>>;
    // Train the model using a input-output pair
    // fn train_once(&mut self, input: &Tensor<T>, output: &Tensor<T>);
}