from torchmetrics.functional import f1
我在尝试从torchmetrics模块导入f1时遇到以下错误。
我安装了torchmetrics模块如下：
!pip install torchmetrics
但问题依然存在。
zte4gxcn1#
也许这段代码会有所帮助
from torchmetrics import F1Score target = torch.tensor([0, 1, 2, 0, 1, 2]) preds = torch.tensor([0, 2, 1, 0, 0, 1]) f1 = F1Score(num_classes=3) f1(preds, target)```
1条答案
按热度按时间zte4gxcn1#
也许这段代码会有所帮助