enchanter.metrics

Classification

calculate_accuracy

enchanter.metrics.calculate_accuracy(inputs: torch.Tensor, targets: torch.Tensor) float[source]

A function that calculates accuracy for batch processing. Returns accuracy as a Python float.

Parameters
  • inputs (torch.Tensor) – shape == [N, n_class]

  • targets (torch.Tensor) – shape == [N]

Returns: accracy (float)