meuron 0.4.0

Meuron is a modular neural network library written in rust for training simple neural networks.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

pub mod binary_cross_entropy;
pub mod cross_entropy;
pub mod mse;
use crate::backend::Backend;
pub use binary_cross_entropy::BinaryCrossEntropy;
pub use cross_entropy::CrossEntropy;
pub use mse::MSE;
use ndarray::{Dimension, RemoveAxis};

pub trait Cost<B: Backend> {
    fn loss<D: Dimension + RemoveAxis>(
        &self,
        predicted: &B::Tensor<D>,
        target: &B::Tensor<D>,
    ) -> f32;

    fn gradient<D: Dimension>(
        &self,
        predicted: &B::Tensor<D>,
        target: &B::Tensor<D>,
    ) -> B::Tensor<D>;
}