We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04481e4 commit 74562dfCopy full SHA for 74562df
chebai/models/base.py
@@ -44,7 +44,7 @@ def __init__(self, loss_cls=None, out_dim=None, **kwargs):
44
self.optimizer_kwargs = kwargs.get("optimizer_kwargs", dict())
45
self.thres = kwargs.get("threshold", 0.5)
46
self.metrics = ["F1Score", "Precision", "Recall", "AUROC"]
47
- self.metric_aggs = ["micro", "macro", "weighted"]
+ self.metric_aggs = ["micro"]
48
for metric in self.metrics:
49
for agg in self.metric_aggs:
50
setattr(
0 commit comments