Skip to content
Snippets Groups Projects

Update content

Merged manli requested to merge github/fork/Bobholamovic/master into master
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
+ 2
2
@@ -41,7 +41,7 @@ class AverageMeter:
# These metrics only for numpy arrays
class Metric(AverageMeter):
__name__ = 'Metric'
def __init__(self, n_classes=2, mode='accum', reduction='binary'):
def __init__(self, n_classes=2, mode='separ', reduction='binary'):
super().__init__(None)
self._cm = AverageMeter(partial(metrics.confusion_matrix, labels=np.arange(n_classes)))
assert mode in ('accum', 'separ')
@@ -94,7 +94,7 @@ class Recall(Metric):
class Accuracy(Metric):
__name__ = 'OA'
def __init__(self, n_classes=2, mode='accum'):
def __init__(self, n_classes=2, mode='separ'):
super().__init__(n_classes=n_classes, mode=mode, reduction='none')
def _compute(self, cm):
return np.nan_to_num(np.diag(cm).sum()/cm.sum())
Loading