kaczmarj commited on
Commit
6afa01d
1 Parent(s): e738b89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -1,3 +1,13 @@
1
  ---
2
  license: cc-by-nc-sa-4.0
 
3
  ---
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-nc-sa-4.0
3
+ library_name: pytorch
4
  ---
5
+
6
+ A prostate cancer tumor classification model. This model is a ResNet34 and was trained on TCGA PRAD.
7
+
8
+ Please refer to the [original GitHub Repo](https://github.com/SBU-BMI/quip_prad_cancer_detection)
9
+
10
+ This model outputs three classes:
11
+ 1. Gleason 3
12
+ 2. Gleason 4 or 5
13
+ 3. Benign