Unverified Commit 816dd603 authored by xuehui's avatar xuehui Committed by GitHub
Browse files

Add Metis Tuner (#534)



* update readme in ga_squad

* update readme

* fix typo

* Update README.md

* Update README.md

* Update README.md

* update readme

* update

* fix path

* update reference

* fix bug in config file

* update nni_arch_overview.png

* update

* update

* update

* add metis tuner code

* 1. fix bug about import 2.update other sdk file

* add auto-gbdt-example and remove unused code

* add metis_tuner into README

* update the README

* update README | remove unused variable

* fix typo

* add sklearn into requirments

* Update src/sdk/pynni/nni/metis_tuner/metis_tuner.py

add default value in __init__
Co-Authored-By: default avatarxuehui1991 <xuehui@microsoft.com>

* Update docs/HowToChooseTuner.md
Co-Authored-By: default avatarxuehui1991 <xuehui@microsoft.com>

* Update docs/HowToChooseTuner.md
Co-Authored-By: default avatarxuehui1991 <xuehui@microsoft.com>

* fix typo | add more comments
parent 573f23ce
...@@ -68,6 +68,16 @@ Optional('tuner'): Or({ ...@@ -68,6 +68,16 @@ Optional('tuner'): Or({
Optional('n_output_node'): int, Optional('n_output_node'): int,
}, },
Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999), Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999),
},{
'builtinTunerName': 'MetisTuner',
'classArgs': {
Optional('optimize_mode'): Or('maximize', 'minimize'),
Optional('no_resampling'): bool,
Optional('no_candidates'): bool,
Optional('selection_num_starting_points'): int,
Optional('cold_start_num'): int,
},
Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999),
},{ },{
'codeDir': os.path.exists, 'codeDir': os.path.exists,
'classFileName': str, 'classFileName': str,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment