diff --git a/src/requirements.txt b/src/requirements.txt index 38d0d35..41891ca 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -1,4 +1,4 @@ numpy==1.17.3 -scikit-learn==0.21.3 +scikit-learn==1.5.0 torch==1.3.0 torchvision==0.4.1 \ No newline at end of file