web-dev-qa-db-fra.com

restore_best_weights émet un arrêt anticipé des keras

J'utilise EarlyStopping de Keras pour mon projet d'apprentissage en profondeur. Les documentations ici mentionne une idée très utile de restaurer les meilleurs poids. Mais je ne suis pas encore en mesure de l'utiliser. J'utilise Keras 2.2.2/TF 1.10, installé à l'aide d'Anaconda. L'appel est simple comme suit. y a-t-il un problème?

es = EarlyStopping(monitor='val_acc', min_delta=1e-4, patience=patience_,verbose=1,restore_best_weights=True)

__init__() a obtenu un argument de mot clé inattendu 'restore_best_weights'

7
Haramoz

Ah, un problème très courant. La documentation en ligne des keras est produite à partir de la branche master dans github, donc les fonctionnalités qui n'ont été introduites que récemment y sont présentes, mais pas encore dans aucune version de Keras.

Alors le restore_best_weights L'option de rappel EarlyStopping a été ajoutée le 18 août 2018, alors que la dernière version de Keras (2.2.2) a été publiée le 28 juillet 2018. Vous devrez donc utiliser Keras master, attendez une version Keras, ou tout simplement pas utiliser l'option.

Mise à jour: Il est maintenant inclus dans la version Keras 2.2. .

14
Matias Valdenegro