Downloads · 30 days
0
BCR_prediciton_model
farzadbz/BCR_prediciton_model
BCR_prediciton_model is a machine learning model from farzadbz. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository provides the pretrained models for biochemical recurrence (BCR) prediction in prostate cancer using:
Access
Public
Updated Nov 15, 2025
Repo size
6.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt3.2 MB · 94%
At a glance
- Access
- Public
- Created
- Nov 15, 2025
- Updated
- Nov 15, 2025
- SHA
- 005c04d3
Model card
From the Hugging Face model README
Multimodal Prostate Cancer BCR Prediction
This repository provides the pretrained models for biochemical recurrence (BCR) prediction in prostate cancer using:
- Multiparametric MRI (mpMRI) features
- Clinical variables
- A multimodal survival model trained with a ranking-based loss (C-index–oriented)
The models are released as PyTorch checkpoints and hosted on Hugging Face for easy reuse.
from huggingface_hub import hf_hub_download
import torch
from src.models import SurvivalModelMM
# Download model
model_path = hf_hub_download(repo_id="farzadbz/BCR_prediction_model", filename="your desired model path")
ckpt = torch.load(model_path, map_location="cpu")
model = SurvivalModelMM(modalities=modalities, d_emb=embed_dim).to(device)
model.load_state_dict(ckpt)
- Created
- Nov 15, 2025
- Updated
- Nov 15, 2025