Tensorflow argmax（）for TFLearn？

Question

我用MNIST数据集的tflearn预测手写的数字。

一切正常，但是我的标签为one_hot 。 tflearn是否有一个函数，与argmax()一样？

Answer 1

您只需执行以下操作即可：

pred = model.predict(test_data)

print([ np.where(r==1)[0][0] for r in np.round(pred) ])

最好。