diff --git a/requirements.txt b/requirements.txt index 650fc46..d34e841 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,4 @@ tensorboard==2.0.0 torch==1.5.0 torchvision==0.6.0 matplotlib==3.2.1 -scikit-learn==0.23.1 +scikit-learn==1.5.0