-
Notifications
You must be signed in to change notification settings - Fork 3
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
Add training/models of Zernike and AreaShape feature space only #44
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Merging this PR is required before I can update figure 3 (PR curves) to the balanced models. |
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.
LGTM! 👍 I thought the R code was super clean and easy to interpret. I left one suggestion for you to address.
Let me know if there are other PRs to review to ensure nothing is blocked.
"ignore", category=ConvergenceWarning, module="sklearn" | ||
# train model on each combination of model type, feature type, and balance type | ||
for balance in balanced_types: | ||
if balance == "balanced": |
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.
I suggest adding a short code comment on why you are specifying the "balanced" model and setting the "unbalanced" model to None
. I am assuming so that it doesn't run this model but I am not 100% sure.
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.
Sounds good! A short comment added in 7f2ca61
I've added the following model training procedures:
I have also updated the evaluation procedures and added a summary visualization