Skip to content

Commit 7ffce00

Browse files
committed
Added skeleton of precision metric to metrics
1 parent 6c2301f commit 7ffce00

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

utils/metrics/precision.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
import torch
2+
import torch.nn as nn
3+
4+
5+
class Precision(nn.Module):
6+
def __init__(self, num_classes):
7+
super().__init__()
8+
9+
self.num_classes = num_classes
10+
11+
def forward(self, y_true, y_pred):
12+
pass

0 commit comments

Comments
 (0)