Tensorflow argmax() pour TFLearn?

Question

J'ai utilisé mnist ensemble de données de tflearn pour prédire les chiffres manuscrits. Tout fonctionne, mais j'ai les étiquettes one_hot. Y at-il une fonction dans tflearn, qui fait la même chose que argmax() dans Tensorflow?

Answer 1

Vous pouvez le faire simplement en faisant ceci:

pred = model.predict(test_data) 

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

Best.