---
license: cc-by-nc-sa-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