metadata
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
This model outputs three classes:
- Gleason 3
- Gleason 4 or 5
- Benign