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

Set the model to evaluation model in notebook #64

Closed
melodiemonod opened this issue Oct 31, 2024 · 0 comments · Fixed by #65
Closed

Set the model to evaluation model in notebook #64

melodiemonod opened this issue Oct 31, 2024 · 0 comments · Fixed by #65
Assignees
Labels
documentation Improvements or additions to documentation

Comments

@melodiemonod
Copy link
Collaborator

The model was not set to evaluation mode in the notebook before evaluating the parameters on test. We included dropout layers. They will randomly set a fraction of the inputs to zero during training, which helps prevent overfitting. However, dropout is also active during evaluation if the model is not set to evaluation mode.

I added weibull_model.eval() and cox_model.eval() before the parameters evaluation on data from dataloader_test.

@melodiemonod melodiemonod self-assigned this Oct 31, 2024
@melodiemonod melodiemonod added the documentation Improvements or additions to documentation label Oct 31, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Improvements or additions to documentation
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant