-
Notifications
You must be signed in to change notification settings - Fork 1
Train code split, Part 3: moving some of the model.py code into sampler.py #9
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
Conversation
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 one is a bit more of a surface level review, as there are pieces to the code that you're likely going to continue busting apart and updated. There are also a few places where the code is weirdly documented or convoluted that we could improve, but it will require a lot of digging. So better to do later after we've done a lot more of the high level stuff.
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.
A few more small comments, but overall this looks good.
…er.py (#9) Moving sampler code into sampler.py. Also moving some functions I found out are for training into train.py.
04d817f Trainer: Changing all literals to enums (#48) 33f9306 Adding ignore for pip 25.2 vulnerability, removing stale ones (#50) c8d30ca Remove mkdocs build dir, add to gitignore (#49) c01121c End-to-end Evaluation Script Example (#45) d231a4a Add Nearest Neighbor Distance Ratio and Epsilon Identifiability Privacy Metrics (#42) 53e423b Adding Mean F1 Score Difference and Hitting Rate Metrics (#39) 91150fd Adding in Hellinger and pMSE metrics (#38) 746644e Tightening Ruff Configuration (#46) 580d55f Adding data_split_ratios to both the diffusion config and the classifier config (#47) 72863be Refactoring core.logger into common.logger and removing it (#41) bef53bf Train code split, Part 4: moving some of the model.py code into dataset.py (#40) 80b0154 Upgrading pip to latest version to solve security issue (#44) 83beba6 New mypy flow and fixes to typing issues that were discovered (#43) 7e77f37 Train code split, Part 3: moving some of the model.py code into sampler.py (#9) git-subtree-dir: deps/midst-toolkit git-subtree-split: 04d817f
PR Type
Fix
Short Description
Clickup Ticket(s): https://app.clickup.com/t/868ew1hg6
Moving sampler code into
sampler.py
. Also moving some functions I found out are for training intotrain.py
.Tests Added
N/A