Skip to main content

Module loss

Module loss 

Source
Expand description

Loss functions (binary cross-entropy, cross-entropy, mean squared error). Loss functions used by the examples and tests.

Modulesยง

bce
Binary cross-entropy. Binary cross-entropy loss.
cross_entropy
Cross-entropy (multi-class). Cross-entropy loss accepting floating-point target distributions.
mse
Mean squared error. Mean squared error loss.