Add model training and evaluation scripts #46
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Starting to build out functionality for Nutrify's data flywheel.
This PR adds
train.py
andevaluate.py
as major scripts for training a FoodVision model as well as evaluating it.There's also a new directory called
foodvision
, this is where all of Nutrify's ML related files will go (for now) when it comes to training a computer vision model for identifying foods.Next will be to create a script to import the "100 most wrong predictions" into Label Studio to update and improve the labels before retraining another model.