jiro_nn 0.5.0

Neural Networks framework with model specification & data preprocessing features.
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

use super::ConvActivationLayer;
use crate::vision::{image::Image, image::ImageTrait};

fn sigmoid(m: &Image) -> Image {
    let exp_neg = m.scalar_mul(-1.).exp();
    let ones = Image::constant(
        m.image_dims().0,
        m.image_dims().1,
        m.channels(),
        m.samples(),
        1.,
    );
    ones.component_div(&(ones.component_add(&exp_neg)))
}

fn sigmoid_prime(m: &Image) -> Image {
    let sig = sigmoid(m);
    let ones = Image::constant(
        m.image_dims().0,
        m.image_dims().1,
        m.channels(),
        m.samples(),
        1.,
    );
    sig.component_mul(&(ones.component_sub(&sig)))
}

pub fn new() -> ConvActivationLayer {
    ConvActivationLayer::new(sigmoid, sigmoid_prime)
}