config_metis.yml 573 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# 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
xuehui's avatar
xuehui committed
15
trialConcurrency: 1
16
maxTrialNumber: 10
xuehui's avatar
xuehui committed
17
tuner:
18
  name: MetisTuner
xuehui's avatar
xuehui committed
19
20
  classArgs:
    optimize_mode: minimize
21
22
trainingService:
  platform: local