DngoTuner.rst 821 Bytes
Newer Older
ChrisZRen's avatar
ChrisZRen committed
1
2
3
DNGO on NNI
===========

4
5
Introduction
------------
ChrisZRen's avatar
ChrisZRen committed
6

7
8
Usage
-----
ChrisZRen's avatar
ChrisZRen committed
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

Installation
^^^^^^^^^^^^

classArgs requirements
^^^^^^^^^^^^^^^^^^^^^^

* **optimize_mode** (*'maximize' or 'minimize'*\ ) - If 'maximize', the tuner will target to maximize metrics. If 'minimize', the tuner will target to minimize metrics.
* **sample_size** (*int, default = 1000*) - Number of samples to select in each iteration. The best one will be picked from the samples as the next trial.
* **trials_per_update** (*int, default = 20*) - Number of trials to collect before updating the model.
* **num_epochs_per_training** (*int, default = 500*) - Number of epochs to train DNGO model.

Example Configuration
^^^^^^^^^^^^^^^^^^^^^

.. code-block:: yaml

   # config.yml
   tuner:
28
     name: DNGOTuner
ChrisZRen's avatar
ChrisZRen committed
29
     classArgs:
30
       optimize_mode: maximize