Wahrscheinlichkeit von Ausgaben für CIFAR erhalten Beispiel (oder ähnlich)

Question

Hey, ich versuche herauszufinden, wie man die Wahrscheinlichkeiten für jede der Klassen im CIFAR-Beispiel erhält. Zur Zeit kann ich die vorhergesagte Klasse in den Eingangstensor eingeben und eine wahr oder falsch zurückbekommen, aber sie sagt mir nicht die Klasse, von der sie denkt, dass sie (von 10) ist und die Wahrscheinlichkeit, die sie auf diese Tatsache legt. Irgendwelche Ideen, wie man diese Informationen nach sess.run() bekommt?

Answer 1

Der CIFAR-10-Beispielcode zur Auswertung (in cifar10_eval.py) erzeugt (als Zwischenausgabe) einen logits Tensor, der Log-Wahrscheinlichkeiten enthält, dass ein bestimmtes Bild in jeder der Klassen enthalten ist. Wenn Sie diesen Tensor an sess.run() übergeben, erhalten Sie die notwendigen Informationen, um Ihnen zu sagen, was die vorhergesagte Klasse für ein Bild ist. Wenn Sie stattdessen tf.nn.softmax(logits) berechnen und diese an sess.run() übergeben, erhalten Sie eine Wahrscheinlichkeit (von 0 bis 1) für jede Klasse.