Module engram::loss

Loss functions.

Enums

• Loss
  Loss functions that can be used to train a neural network.

Functions

• bce
  Returns the binary cross entropy between the predictions and the targets.
• d_bce
• d_mse
• mse
  Returns mean squared error between the predictions and the targets.