config_metis.yml 573 Bytes
Newer Older
qianyj's avatar
qianyj committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# The search space of Metis tuner is slightly different from TPE and others.
# See Metis tuner' doc for details: https://nni.readthedocs.io/en/stable/Tuner/MetisTuner.html
searchSpace:
  num_leaves:
    _type: choice
    _value: [31, 28, 24, 20]
  learning_rate:
    _type: choice
    _value: [0.01, 0.05, 0.1, 0.2]
  bagging_freq:
    _type: choice
    _value: [1, 2, 4, 8, 10]

trialCommand: python3 main.py
trialConcurrency: 1
maxTrialNumber: 10
tuner:
  name: MetisTuner
  classArgs:
    optimize_mode: minimize
trainingService:
  platform: local