Trait rusty_machine::learning::toolkit::activ_fn::ActivationFunc [−] [src]

pub trait ActivationFunc {
    fn func(x: f64) -> f64;
    fn func_grad(x: f64) -> f64;
    fn func_inv(x: f64) -> f64;
}

Trait for activation functions in models.

Required Methods

fn func(x: f64) -> f64

The activation function.

fn func_grad(x: f64) -> f64

The gradient of the activation function.

fn func_inv(x: f64) -> f64

The inverse of the activation function.

Implementors