sentence_representation_matching
sentence_representation_matching copied to clipboard
sbert发现一处bug?
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())
吗?
感谢指正🤝