ML Program to Train and create a Binary Classifier Model using Transfer Learning. Make sure to train them on your own LABELLED Dataset
Getting the perfect Dataset is a huge challenge and also figuring out the dependancies to run the ML trainer.
I would recommend to use conda environment and install packages using conda (avoid using pip)
Note: Both models below were trained on different datasets
Model_0
Epochs: 3
Accuracy: 99.681 0%
Precision: 97.9730 %
F1 Score: 95.0820 %
Here's the attached training data for model0:
Model_1
Epochs: 10
Accuracy: 96.5472 %
Precision: 96.5481 %
F1 Score: 96.5229 %
Here's the attached training data for model1: