primetimetran commited on
Commit
47d4b6c
1 Parent(s): 67daf24
Files changed (3) hide show
  1. .gitignore +1 -0
  2. README.md +56 -0
  3. models/HBF_KNN_Classifier.pickle +3 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .vscode
README.md CHANGED
@@ -1,3 +1,59 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - python
5
+ thumbnail: "url to a thumbnail used in social sharing"
6
+ tags:
7
+ - DICOM
8
+ - cancer
9
+ - medical
10
+ - medical imaging
11
+ - classification
12
+ datasets:
13
+ - [dataset1](https://www.breastcancer.org/facts-statistics](https://www.cancerimagingarchive.net/collection/breast-cancer-screening-dbt)
14
+ metrics:
15
+ - 62% Sensitivity
16
+ base_model: "N/A"
17
  ---
18
+
19
+ # HerBreastsFriend(HBF)
20
+
21
+ A model for identifying breast cancer in patients inspired by a study conducted by Duke & blogged about by jamanetwork[^1].
22
+
23
+ Their studies finding's were that there's a lot of room for improvement. They came to this conclusion after building their
24
+ own AI model for breast cancer detection/prognoses and achieved a 65% on sensitivity.
25
+
26
+ ### Details
27
+
28
+ - KNN strategy
29
+ - n_neighbors=5
30
+ - StandardScaler
31
+ - PCA
32
+ - n_components=2
33
+ - Trained on limited dataset(1997 images)
34
+ - I had to limit the number of data points in my model because my machine kept freezing. WIP on a solution.
35
+ - Hosted by the amazing cancerimagingarchive[^2]
36
+
37
+ ### Classification Report
38
+
39
+ The initial release of HBF scored the following in our classification. 62% for average weighted across all features. A lot of room for improvement.
40
+
41
+ ```sh
42
+ precision recall f1-score support
43
+
44
+ Normal 0 0.62 0.80 0.70 956
45
+ Actionable 1 0.61 0.58 0.59 760
46
+ Benign 2 0.69 0.07 0.12 164
47
+ Cancer 3 0.47 0.08 0.13 117
48
+
49
+ accuracy 0.62 1997
50
+ macro avg 0.60 0.38 0.39 1997
51
+ weighted avg 0.61 0.62 0.58 1997
52
+ ```
53
+
54
+ ### FAQ
55
+
56
+ I'm considering making this open source. If you'd like to contribute please give a star to let me know there's others interested.
57
+
58
+ [^1] Duke Study https://jamanetwork.com/journals/jamanetworkopen/fullarticle/2783046
59
+ [^2] [cancerimagingarchive https://www.breastcancer.org/facts-statistics](https://www.cancerimagingarchive.net/collection/breast-cancer-screening-dbt)
models/HBF_KNN_Classifier.pickle ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18ddaa1ab4fb88645e16a8fc4940e4ee14d6ae649cd36b2443161ca9da4cf359
3
+ size 3141041773