changed sklearn to scikit-learn

benchmarks/data.py

@@ -14,8 +14,8 @@
 import PIL
 import torch
 import torch.nn.functional as F
-from sklearn.datasets import make_classification
-from sklearn.model_selection import train_test_split
+from scikit-learn.datasets import make_classification
+from scikit-learn.model_selection import train_test_split
 from torchvision import transforms
 
 

benchmarks/run_historical_benchmarks.py

@@ -90,7 +90,7 @@ def get_dates(day=26):
 
 # install PyTorch
 subprocess.call(
-    f"{venv} pip3 install onnx==1.6.0 tensorboard pandas sklearn", shell=True
+    f"{venv} pip3 install onnx==1.6.0 tensorboard pandas scikit-learn", shell=True
 )
 stable_url = "https://download.pytorch.org/whl/torch_stable.html"
 pip_torch = f"pip install torch==1.5.1+cu{cuda_version} torchvision==0.6.1+cu{cuda_version} -f https://download.pytorch.org/whl/torch_stable.html"

requirements.txt

@@ -7,4 +7,4 @@ pyyaml>=5.3.1
 tensorboard
 future
 scipy>=1.6.0
-sklearn
+scikit-learn