Dazu habe ich versucht, "tf.argmax (pred, 1)", und das Ergebnis ist [1 3 4 1 2].Faltung neuronales Netzwerk prognostiziert die Wahrscheinlichkeit einer Klasse
Wie die Wahrscheinlichkeit der Prognosekategorie kennen? Bietet Tensorflow API?
Was Sie tun können, ist die Argmax der Vorhersage zu bekommen, und erhalten Sie die Vorhersage bei diesem Index. Diese ist alles einfacher in numpy (zum Beispiel die amax Funktion: https://docs.scipy.org/doc/numpy/reference/generated/numpy.amax.html)
tfpredicted,loss, acc = sess.run([pred,cost, accuracy], feed_dict={x: batch_x,y: batch_y,keep_prob: 1.})
print(np.amax(tfpredicted))
Ich habe den Code nicht testen, aber hoffen, dass es funktioniert. Sonst müssen Sie vielleicht die max Funktionen numpy Angebote betrachten.
Viel Glück!
Also meinst du das Ergebnis von "pred" ist alle Label Wahrscheinlichkeit. Aber das Ergebnis ist [[-11511043. 5100159. 5619258. -1942 ...] [....]]. Wie zur Wahrscheinlichkeit konvertieren? – clout
keine Wahrscheinlichkeit, aber Aktivierung ... hast du dein Netzwerk bereits trainiert? Und Sie könnten eine Softmax-Funktion verwenden, um alles zwischen 0 und 1 zu bekommen;) – rmeertens
Ich dachte, es war 0 zu 1, aber das Ergebnis ist [3365887. 4442517. 0. 4042083.25 0.]. Ich benutze dieses Beispiel (tf.nn.softmax_cross_entropy_with_logits) – clout