我试图使用f1数据集计算多类分类问题的Cifar10评分。我正在从学习库中导入f1流体力学。但是,我一直收到以下错误消息:
ValueError: Classification metrics can't handle a mix of multiclass and continuous-multioutput targets
下面是我在验证集上测试模型的功能。有人能解释在执行多类分类时如何计算f1吗?我有点糊涂了。
@torch.no_grad()
def valid_function(model, optimizer, val_loader):
model.eval()
val_loss = 0.0
val_accu = 0.0
f_one = []
for i, (x_val, y_val) in enumerate(val_loader):
x_val, y_val = x_val.to(device), y_val.to(device)
val_pred = model(x_val)
loss = criterion(val_pred, y_val)
val_loss += loss.item()
val_accu += accuracy(val_pred, y_val)
f_one.append(f1_score(y_val.cpu(), val_pred.cpu()))
val_loss /= len(val_loader)
val_accu /= len(val_loader)
print('Val Loss: %.3f | Val Accuracy: %.3f'%(val_loss,val_accu))
return val_loss, val_accu
发布于 2022-04-18 21:37:42
问题在于:
val_pred = model(x_val)
您需要转换加载模型的方式。例如,在您的案例中:
val_pred = np.argmax(model.predict(x_val), axis=-1)
https://stackoverflow.com/questions/71874695
复制相似问题