[go: nahoru, domu]

Skip to content
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

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

lpizzinidev
Copy link
Contributor
@lpizzinidev lpizzinidev commented Feb 29, 2024

Updates the "Review Classification using Active Learning" example to support Keras v3.

@@ -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"
Copy link
Contributor Author

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)
Copy link
Contributor Author

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.

@lpizzinidev
Copy link
Contributor Author

@fchollet ✋ Can you please review?

Copy link
Member
@fchollet fchollet left a 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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants