pytorch求多分类的准确率

accracy = np.mean((torch.argmax(out,1)==torch.argmax(y,1)).numpy())

猜你喜欢

转载自blog.csdn.net/weixin_38241876/article/details/90605198