Skip to content

Latest commit

 

History

History
13 lines (7 loc) · 1.15 KB

LEARN.md

File metadata and controls

13 lines (7 loc) · 1.15 KB

This is a Breast Cancer Classification Model based on fine-tuned VGG-16 and SVM to classify the INbreast Dataset

I have used a whole image model by @lishen, trained it on Google Colab and added an SVM.svc classifier on top.

Mr. Li Shen's model had performed great with AUC=0.95 (check his Repo and his Paper 🔥);

I managed to get AUC=0.98 with SVM and even better with MLP and Random Forests:

AUC-results

For more details, check full document HERE 📌

🚨 For any bugs or questions, feel free to report an issue 🚩