Skip to content

Commit 74562df

Browse files
committed
Remove obsolete metrics
1 parent 04481e4 commit 74562df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/models/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def __init__(self, loss_cls=None, out_dim=None, **kwargs):
4444
self.optimizer_kwargs = kwargs.get("optimizer_kwargs", dict())
4545
self.thres = kwargs.get("threshold", 0.5)
4646
self.metrics = ["F1Score", "Precision", "Recall", "AUROC"]
47-
self.metric_aggs = ["micro", "macro", "weighted"]
47+
self.metric_aggs = ["micro"]
4848
for metric in self.metrics:
4949
for agg in self.metric_aggs:
5050
setattr(

0 commit comments

Comments
 (0)