diff --git a/requirements.txt b/requirements.txt index d95d59a..d761bf9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,4 @@ torch==1.4.0 ordered_set==3.1 numpy==1.16.2 torch_scatter==2.0.4 -scikit_learn==0.21.1 +scikit_learn==1.5.0