primetimetran
commited on
Commit
•
47d4b6c
1
Parent(s):
67daf24
Update
Browse files- .gitignore +1 -0
- README.md +56 -0
- 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
|