From faf59303b9d8f94914957e3fe1da0fd9a8f26b9e Mon Sep 17 00:00:00 2001 From: Amithabh-A Date: Mon, 17 Jun 2024 19:22:18 +0530 Subject: [PATCH] changed sklearn to scikit-learn --- benchmarks/data.py | 4 ++-- benchmarks/run_historical_benchmarks.py | 2 +- requirements.txt | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/benchmarks/data.py b/benchmarks/data.py index 444a4899..d514326e 100644 --- a/benchmarks/data.py +++ b/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 diff --git a/benchmarks/run_historical_benchmarks.py b/benchmarks/run_historical_benchmarks.py index dc4eea89..1d955b98 100644 --- a/benchmarks/run_historical_benchmarks.py +++ b/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" diff --git a/requirements.txt b/requirements.txt index 140fdf8d..47258683 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ pyyaml>=5.3.1 tensorboard future scipy>=1.6.0 -sklearn +scikit-learn