This directory contains the loss functions used in common machine learning tasks, such as cross entropy loss, and mean squared error loss. For specific ML task, user can define their own loss function in a similar way by creating a child class of nn.Module and defining the forward(prediction, target) function to compute the loss.
losses
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||