Dataset : https://www.kaggle.com/uciml/breast-cancer-wisconsin-data
Detect stage of cancer from benign or malignant.
Best 93.96% accuracy by DecisionTreeClassifier of scikit learn.
96% accuracy by simple 1 hidden layer neural net. Trained with 100 epochs. Having more hidden layers may increase accuracy a little. But it may not generalize and overfit the train data.
Tried several models on the same dataset. I have also tried ensambling of several models, but it turns out that simple DecisionTreeClassifier beats them.
-
Notifications
You must be signed in to change notification settings - Fork 3
prashant-kikani/breast-cancer-detection
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Detects stage of breast cancer.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published