rustyml 0.16.0

A high-performance machine learning & deep learning library in pure Rust, offering ML algorithms and neural network support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Noise-injection regularization layers
//!
//! Groups the 2 noise-injection layers and re-exports them. Multiplicative
//! [`GaussianDropout`](crate::neural_network::layers::GaussianDropout) scales inputs by
//! `N(1, rate/(1 - rate))` during training. Additive
//! [`GaussianNoise`](crate::neural_network::layers::GaussianNoise) adds zero-mean `N(0, stddev^2)`
//! noise during training. Both are identity maps at inference.
//!
//! This file defines no shared infrastructure of its own. The 2 layers read the training flag
//! of the [`Ctx`](crate::neural_network::Ctx), and they reuse the validation helpers from the
//! parent [`regularization`](crate::neural_network::layers::regularization) module

/// Gaussian Dropout layer for neural networks
pub mod gaussian_dropout;
/// Gaussian Noise layer for neural networks
pub mod gaussian_noise;

pub use gaussian_dropout::GaussianDropout;
pub use gaussian_noise::GaussianNoise;