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
31
32
//! Provides methods to calculate the loss (cost) of some output.
//!
//! A loss function is also sometimes called cost function.

/// macro helper for default loss
#[macro_export]
macro_rules! impl_ilayer_loss {
    () => {
        fn exact_num_output_blobs(&self) -> Option<usize> {
            Some(1)
        }
        fn exact_num_input_blobs(&self) -> Option<usize> {
            Some(1)
        }
        fn auto_output_blobs(&self) -> bool {
            true
        }

        fn loss_weight(&self, output_id: usize) -> Option<f32> {
            if output_id == 0 {
                Some(1f32)
            } else {
                None
            }
        }
    };
}

pub use self::mean_squared_error::MeanSquaredError;
pub use self::negative_log_likelihood::{NegativeLogLikelihood, NegativeLogLikelihoodConfig};
pub mod mean_squared_error;
pub mod negative_log_likelihood;