diff --git a/requirements.txt b/requirements.txt index eb4e335..9b8cb4b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,4 @@ pandas>=1.0.0 tqdm>=4.0.0 scikit-learn>=0.18 torchvision>=0.7.0 +scikit-survival>=0.15.0