Skip to content

ML Program to Train and create a Binary Classifier Model using Transfer Learning. Make sure to train them on your own LABELLED Dataset

Notifications You must be signed in to change notification settings

41vi4p/ML_Binary_Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML_Binary_Classifier

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

Models Trained using this Trainer

Model_0

Epochs: 3
Accuracy: 99.681 0%
Precision: 97.9730 %
F1 Score: 95.0820 %
Here's the attached training data for model0:
Screenshot from 2024-09-07 18-50-59

Model_1

Epochs: 10
Accuracy: 96.5472 %
Precision: 96.5481 %
F1 Score: 96.5229 %
Here's the attached training data for model1:
Training Data Model_1

About

ML Program to Train and create a Binary Classifier Model using Transfer Learning. Make sure to train them on your own LABELLED Dataset

Resources

Stars

Watchers

Forks