CASurv

Code for paper: "Censor-aware Semi-supervised Learning for Survival Time Prediction from Medical Images" [arxiv].

Install

pip install git+https://github.com/renato145/CASurv.git

Docker

Check: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker to use the --gpus flag.

# Build the image:
docker build -t casurv .
# Run the image:
docker run --rm -it --gpus all -p 8888:8888 -v DATA_PATH:/workspace/notebooks/data casurv:latest