tumourkit.segmentation.hovernet.models.hovernet.utils.xentropy_loss
- tumourkit.segmentation.hovernet.models.hovernet.utils.xentropy_loss(true, pred, reduction='mean', device='cuda')
Cross entropy loss. Assumes NHWC!
- Args:
pred: prediction array true: ground truth array
- Returns:
cross entropy loss