diff --git a/requirements.txt b/requirements.txt index 8a64d42..cae479e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,6 +6,6 @@ optuna>=1.0.0 pandas pyarrow seaborn -sklearn +scikit-learn tqdm transformers[ja] diff --git a/setup.py b/setup.py index f2eeb89..f9faf51 100644 --- a/setup.py +++ b/setup.py @@ -37,7 +37,7 @@ setup( 'pandas', 'pyarrow', 'seaborn', - 'sklearn', + 'scikit-learn', 'tqdm', 'transformers>=2.3.0', ],