#6 Add another stage for modeling using ResNet50V2

Merged
nirbarazida wants to merge 7 commits from nirbarazida:task-3-ResNet50V2 into nirbarazida:master
Nir Barazida created the pull request 2 months ago
There is no content yet.
Nir Barazida commented 2 months ago
Owner

TODO:

TODO: - Change the name of the model dir in task-3 to model-checkpoint & update it in the [const file R14](https://dagshub.com/nirbarazida/Pneumonia-Classification-Monorepo/pulls/6/files/task-3-modeling/src/const/modeling_const.py?page=0#diff-8da91d6413282764a706cb5384ff02551d2f09fR14) - Track the prod_model with DVC - change to binary: - [task-3-modeling/src/train.py R33] (https://dagshub.com/nirbarazida/Pneumonia-Classification-Monorepo/pulls/6/files/task-3-modeling/src/train.py?page=0#diff-16bfb924484fec6a8d29ee9453650ef0a0817baR33) change it to `sigmoid` with `num class = 1` - [task-3-modeling/src/train.py R45](https://dagshub.com/nirbarazida/Pneumonia-Classification-Monorepo/pulls/6/files/task-3-modeling/src/train.py?page=0#diff-16bfb924484fec6a8d29ee9453650ef0a0817baR45) - change it to `binary_crossentropy`
Nir Barazida referenced this issue from a commit 2 months ago
This pull request has been merged successfully!
Sign in to join this conversation.
No Milestone
No assignee
1 Participants
Loading...
Cancel
Save
There is no content yet.