Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ignore_index to Accuracy metric #155

Closed
wants to merge 14 commits into from
Prev Previous commit
Next Next commit
Add ignore_index to accuracy (PyTorchLightning#61)
  • Loading branch information
avinashsai authored Apr 3, 2021
commit 5132bc0a93ab02b3d0734608376a83bfb2e094dd
2 changes: 1 addition & 1 deletion torchmetrics/functional/classification/accuracy.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,9 @@
import torch
from torch import Tensor, tensor

from torchmetrics.functional.classification.stat_scores import _del_column
from torchmetrics.utilities.checks import _input_format_classification
from torchmetrics.utilities.enums import DataType
from torchmetrics.functional.classification.stat_scores import _del_column


def _accuracy_update(
Expand Down