pytorch求多分类的准确率

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

你可能感兴趣的:(人工智能,深度学习,pytorch)