registered_algorithms.yml 3.11 KB
Newer Older
1
2
advisors:
- builtinName: Hyperband
3
4
  classArgsValidator: nni.algorithms.hpo.hyperband_advisor.HyperbandClassArgsValidator
  className: nni.algorithms.hpo.hyperband_advisor.Hyperband
5
6
  source: nni
- builtinName: BOHB
7
8
  classArgsValidator: nni.algorithms.hpo.bohb_advisor.BOHBClassArgsValidator
  className: nni.algorithms.hpo.bohb_advisor.BOHB
9
10
11
  source: nni
assessors:
- builtinName: Medianstop
12
13
  classArgsValidator: nni.algorithms.hpo.medianstop_assessor.MedianstopClassArgsValidator
  className: nni.algorithms.hpo.medianstop_assessor.MedianstopAssessor
14
15
  source: nni
- builtinName: Curvefitting
16
17
  classArgsValidator: nni.algorithms.hpo.curvefitting_assessor.CurvefittingClassArgsValidator
  className: nni.algorithms.hpo.curvefitting_assessor.CurvefittingAssessor
18
19
20
  source: nni
tuners:
- builtinName: PPOTuner
21
22
  classArgsValidator: nni.algorithms.hpo.ppo_tuner.PPOClassArgsValidator
  className: nni.algorithms.hpo.ppo_tuner.PPOTuner
23
24
  source: nni
- builtinName: SMAC
25
26
  classArgsValidator: nni.algorithms.hpo.smac_tuner.SMACClassArgsValidator
  className: nni.algorithms.hpo.smac_tuner.SMACTuner
27
28
29
30
  source: nni
- builtinName: TPE
  classArgs:
    algorithm_name: tpe
31
32
  classArgsValidator: nni.algorithms.hpo.hyperopt_tuner.HyperoptClassArgsValidator
  className: nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner
33
34
35
36
37
  source: nni
- acceptClassArgs: false
  builtinName: Random
  classArgs:
    algorithm_name: random_search
38
39
  classArgsValidator: nni.algorithms.hpo.hyperopt_tuner.HyperoptClassArgsValidator
  className: nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner
40
41
42
43
  source: nni
- builtinName: Anneal
  classArgs:
    algorithm_name: anneal
44
45
  classArgsValidator: nni.algorithms.hpo.hyperopt_tuner.HyperoptClassArgsValidator
  className: nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner
46
47
  source: nni
- builtinName: Evolution
48
49
  classArgsValidator: nni.algorithms.hpo.evolution_tuner.EvolutionClassArgsValidator
  className: nni.algorithms.hpo.evolution_tuner.EvolutionTuner
50
51
52
  source: nni
- acceptClassArgs: false
  builtinName: BatchTuner
53
  className: nni.algorithms.hpo.batch_tuner.BatchTuner
54
55
56
  source: nni
- acceptClassArgs: false
  builtinName: GridSearch
57
  className: nni.algorithms.hpo.gridsearch_tuner.GridSearchTuner
58
59
  source: nni
- builtinName: NetworkMorphism
60
61
  classArgsValidator: nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismClassArgsValidator
  className: nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner
62
63
  source: nni
- builtinName: MetisTuner
64
65
  classArgsValidator: nni.algorithms.hpo.metis_tuner.MetisClassArgsValidator
  className: nni.algorithms.hpo.metis_tuner.MetisTuner
66
67
  source: nni
- builtinName: GPTuner
68
69
  classArgsValidator: nni.algorithms.hpo.gp_tuner.GPClassArgsValidator
  className: nni.algorithms.hpo.gp_tuner.GPTuner
70
71
  source: nni
- builtinName: PBTTuner
72
73
  classArgsValidator: nni.algorithms.hpo.pbt_tuner.PBTClassArgsValidator
  className: nni.algorithms.hpo.pbt_tuner.PBTTuner
74
75
  source: nni
- builtinName: RegularizedEvolutionTuner
76
77
  classArgsValidator: nni.algorithms.hpo.regularized_evolution_tuner.EvolutionClassArgsValidator
  className: nni.algorithms.hpo.regularized_evolution_tuner.RegularizedEvolutionTuner
78
  source: nni