Open Garrfiiel opened 7 months ago
sbert/train.py compute_acc()函数 23行accuracy_score() pred和lebel入参好像反了？ acc = accuracy_score(pred.detach().cpu().numpy(), label.cpu().numpy()) 应该是acc = accuracy_score(label.cpu().numpy()， pred.detach().cpu().numpy())吗？
acc = accuracy_score(pred.detach().cpu().numpy(), label.cpu().numpy())
acc = accuracy_score(label.cpu().numpy()， pred.detach().cpu().numpy())
感谢指正🤝
Macielyoung
commented
7 months ago Macielyoung
commented
7 months ago
sbert/train.py compute_acc()函数 23行accuracy_score() pred和lebel入参好像反了？
acc = accuracy_score(pred.detach().cpu().numpy(), label.cpu().numpy())应该是acc = accuracy_score(label.cpu().numpy()， pred.detach().cpu().numpy())吗？