Quelle est la méthode du générateur pour la prédiction de probabilité dans Keras?

Question

Je veux prédire les probabilités d'un problème de bi-classification. Auparavant, j'utilisais model.predict_proba ou predict_on_batch pour ce problème. Maintenant, je veux utiliser des générateurs dans mes scripts, mais je ne trouve pas de générateur tel que evaluate_generator ou predict_generator. Evalue_generator ou predict_generator ne généreront pas de probabilités. Quelle est la méthode du générateur pour la prédiction de probabilité dans Keras?

Answer 1

Quel que soit le résultat est une probabilité ou non dépend du modèle de réseau neuronal réel, pas sur predict_generator. Si votre modèle génère déjà des probabilités, c'est-à-dire qu'il a une activation softmax en sortie, alors utiliser predict_generator devrait vous donner des valeurs de probabilité.