Use scikit-learn in extra dependencies. (#5310)

2020-02-15 07:12:51 +08:00
parent c35cdecddd
commit 8ca9744b07
1 changed files with 1 additions and 1 deletions
--- a/python-package/setup.py
+++ b/python-package/setup.py
@@ -288,7 +288,7 @@ if __name__ == '__main__':
          },
          extras_require={
              'pandas': ['pandas'],
-              'sklearn': ['sklearn'],
+              'scikit-learn': ['scikit-learn'],
              'dask': ['dask', 'pandas', 'distributed'],
              'datatable': ['datatable'],
              'plotting': ['graphviz', 'matplotlib']