Struct rusty_machine::learning::nnet::BCECriterion

pub struct BCECriterion;

The binary cross entropy criterion.

Uses the Sigmoid activation function and the cross entropy error.

Trait Implementations

impl Criterion for BCECriterion

type ActFunc = Sigmoid

The activation function for the criterion.

type Cost = CrossEntropyError

The cost function for the criterion.

fn activate(&self, mat: Matrix<f64>) -> Matrix<f64>

The activation function applied to a matrix.

fn grad_activ(&self, mat: Matrix<f64>) -> Matrix<f64>

The gradient of the activation function applied to a matrix.

fn cost(&self, outputs: &Matrix<f64>, targets: &Matrix<f64>) -> f64

The cost function.

fn cost_grad(&self, outputs: &Matrix<f64>, targets: &Matrix<f64>) -> Matrix<f64>

The gradient of the cost function.