mwptoolkit.loss.cross_entropy_loss

class mwptoolkit.loss.cross_entropy_loss.CrossEntropyLoss(weight=None, mask=None, size_average=True)[source]

Bases: AbstractLoss

Parameters

  • weight (Tensor, optional) – a manual rescaling weight given to each class.

  • mask (Tensor, optional) – index of classes to rescale weight

eval_batch(outputs, target)[source]

calculate loss

Parameters

  • outputs (Tensor) – output distribution of model.

  • target (Tensor) – target classes.

get_loss()[source]

return loss

Returns

loss (float)