| license: cc-by-4.0 | |
| library_name: pytorch | |
| A prostate cancer tumor classification model. This model is a ResNet34 and was trained on TCGA PRAD. | |
| Please refer to the [original GitHub Repo](https://github.com/SBU-BMI/quip_prad_cancer_detection) | |
| This model outputs three classes: | |
| 1. Gleason 3 | |
| 2. Gleason 4 or 5 | |
| 3. Benign |