2017-08-30 3 views
0

Ich möchte Wahrscheinlichkeiten für ein bi-Klassifikations-Problem vorhersagen. Zuvor habe ich model.predict_proba oder predict_on_batch für dieses Problem verwendet. Jetzt möchte ich Generatoren in meinen Skripten verwenden, aber ich kann keinen Generator wie evaluate_generator oder predict_generator finden. Sowohl evaluate_generator als auch predict_generator erzeugen keine Wahrscheinlichkeiten. Was ist die Generatormethode für die Wahrscheinlichkeitsvorhersage in Keras?Was ist die Generatormethode für die Wahrscheinlichkeitsvorhersage in Keras?

+0

Welche Keras-Version verwenden Sie? es scheint, dass Keras 'evaluate_generator' und' predicate_generator' hat, und der 'predicate_generator' gibt ein Numpy-Array von Vorhersagen (Wahrscheinlichkeit) zurück. Siehe [Keras Sequential] (https://keras.io/models/sequential/) –

Antwort

0

Unabhängig davon, ob die Ausgabe eine Wahrscheinlichkeit ist oder nicht, hängt von dem tatsächlichen neuronalen Netzwerkmodell ab, nicht von predict_generator. Wenn Ihr Modell bereits Wahrscheinlichkeiten ausgibt, dh, es hat eine Softmax-Aktivierung am Ausgang, dann sollte die Verwendung von predict_generator Ihnen Wahrscheinlichkeitswerte geben.

Verwandte Themen