Losses

object Losses(source)

Java-friendly factory for loss functions.

Example usage from Java:

Loss loss = Losses.crossEntropy();
Loss mse = Losses.mse();
Loss bce = Losses.binaryCrossEntropy(1e-7f);

Functions

Link copied to clipboard

Binary cross-entropy with logits (numerically stable).

Link copied to clipboard

Binary cross-entropy loss (predictions should be probabilities in 0, 1).

Link copied to clipboard

Categorical cross-entropy (alias for crossEntropy).

Link copied to clipboard

Cross-entropy loss (combines log-softmax with NLL).

Link copied to clipboard
fun hinge(margin: Float = 1.0f): Loss

Hinge loss for SVM-style classification.

Link copied to clipboard
fun huber(delta: Float = 1.0f): Loss

Huber (Smooth L1) loss — quadratic for small errors, linear for large.

Link copied to clipboard

Mean Absolute Error loss.

Link copied to clipboard

Mean Squared Error loss.

Link copied to clipboard
fun poisson(logInput: Boolean = true, epsilon: Float = 1.0E-8f): Loss

Poisson negative log-likelihood loss for count data.

Link copied to clipboard

Sparse categorical cross-entropy (integer target indices).

Link copied to clipboard

Squared hinge loss.