Skip to content

Commit

Permalink
Merge pull request #182 from microsoft/master
Browse files Browse the repository at this point in the history
Add sklearn installation in setup.py (microsoft#1157)
  • Loading branch information
SparkSnail authored Jun 10, 2019
2 parents cbf88f7 + ce2d8d9 commit 0235102
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
3 changes: 2 additions & 1 deletion deployment/pypi/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@
'numpy',
'scipy',
'coverage',
'colorama'
'colorama',
'sklearn'
],
classifiers = [
'Programming Language :: Python :: 3',
Expand Down
4 changes: 0 additions & 4 deletions docs/en_US/BuiltinTuner.md
Original file line number Diff line number Diff line change
Expand Up @@ -302,10 +302,6 @@ tuner:

Note that the only acceptable types of search space are `choice`, `quniform`, `uniform` and `randint`.

**Installation**

Metis Tuner requires [sklearn](https://scikit-learn.org/), so users should install it first. User could use `pip3 install sklearn` to install it.

**Suggested scenario**

Similar to TPE and SMAC, Metis is a black-box tuner. If your system takes a long time to finish each trial, Metis is more favorable than other approaches such as random search. Furthermore, Metis provides guidance on the subsequent trial. Here is an [example](https://github.com/Microsoft/nni/tree/master/examples/trials/auto-gbdt/search_space_metis.json) about the use of Metis. User only need to send the final result like `accuracy` to tuner, by calling the nni SDK. [Detailed Description](./MetisTuner.md)
Expand Down
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,8 @@ def read(fname):
'scipy',
'schema',
'PythonWebHDFS',
'colorama'
'colorama',
'sklearn'
],

entry_points = {
Expand Down

0 comments on commit 0235102

Please sign in to comment.