-
Notifications
You must be signed in to change notification settings - Fork 2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Updated Review Classification using Active Learning example for Keras v3 #1778
base: master
Are you sure you want to change the base?
Conversation
@@ -281,15 +282,19 @@ def train_full_model(full_train_dataset, val_dataset, test_dataset): | |||
], | |||
) | |||
|
|||
# We will save the best model at every epoch and load the best one for evaluation on the test set | |||
# We will save the weights of the best model at every epoch and load the best ones for evaluation on the test set | |||
checkpoint_file = "FullModelCheckpoint.weights.h5" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Saving weights because of this issue.
# Getting predictions from previously trained model | ||
predictions = model.predict(test_dataset) | ||
|
||
# Generating labels from the output probabilities | ||
rounded = tf.where(tf.greater(predictions, 0.5), 1, 0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This value was not used.
@fchollet ✋ Can you please review? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking good! Please add the generated files. Sorry for the delayed review!
Updates the "Review Classification using Active Learning" example to support Keras v3.