diff --git a/examples/image-classification/requirements.txt b/examples/image-classification/requirements.txt index 4f22f39c4..cf8e1a5a6 100644 --- a/examples/image-classification/requirements.txt +++ b/examples/image-classification/requirements.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cpu torchvision==0.15.2 datasets>=1.15.0 -scikit-learn==0.24.2 +scikit-learn==1.0.1 evaluate