diff --git a/setup.py b/setup.py index bd49278..3d93b64 100644 --- a/setup.py +++ b/setup.py @@ -39,7 +39,7 @@ def read(fname): install_requires = [ 'numpy', 'pandas', - 'sklearn', + 'scikit-learn', ] @@ -65,4 +65,4 @@ def read(fname): 'PseudoLabelSSL': ['keras'], }, keywords=['semi-supervised learning','ssl', 'deep learning', 'machine learning', 'semisupervised'] - ) \ No newline at end of file + )