diff --git a/requirements.txt b/requirements.txt index 47c2fd6..a18655f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ GPUtil==1.4.0 numpy==1.18.4 -scikit-learn==0.23.0 +scikit-learn==1.5.0 torch==1.5.0 \ No newline at end of file