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:

  1. Gleason 3
  2. Gleason 4 or 5
  3. Benign
Downloads last month
256
Safetensors
Model size
21.3M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported third-party Inference Providers, and HF Inference API was unable to determine this model’s pipeline type.

Collection including kaczmarj/prostate-tumor-resnet34.tcga-prad