rocreguant/checkpoint_roc_auc.py

Created January 12, 2021 20:45

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/rocreguant/7122cac139593e56d9715ea61671b9c8.js"></script>
Save rocreguant/7122cac139593e56d9715ea61671b9c8 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

checkpoint_roc_auc.py

checkpoint = ModelCheckpoint(weights_path, monitor="auc_roc", verbose=1, save_best_only=True, mode='max', save_weights_only = True)

Author

rocreguant commented Jan 12, 2021

Create a ckeckpoint in keras / tensorflow using a self customized metric. The criteria is ROC AUC.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment