naive_tuner.py 1.27 KB
Newer Older
liuzhe-lz's avatar
liuzhe-lz committed
1
2
3
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

Deshui Yu's avatar
Deshui Yu committed
4
5
import json
import logging
QuanluZhang's avatar
QuanluZhang committed
6
import os
Deshui Yu's avatar
Deshui Yu committed
7
8

from nni.tuner import Tuner
9
from nni.utils import extract_scalar_reward
Deshui Yu's avatar
Deshui Yu committed
10
11
12

_logger = logging.getLogger('NaiveTuner')
_logger.info('start')
QuanluZhang's avatar
QuanluZhang committed
13
14
15

_pwd = os.path.dirname(__file__)
_result = open(os.path.join(_pwd, 'tuner_result.txt'), 'w')
Deshui Yu's avatar
Deshui Yu committed
16
17

class NaiveTuner(Tuner):
18
    def __init__(self, optimize_mode):
Deshui Yu's avatar
Deshui Yu committed
19
20
21
        self.cur = 0
        _logger.info('init')

22
    def generate_parameters(self, parameter_id, **kwargs):
Deshui Yu's avatar
Deshui Yu committed
23
24
25
26
        self.cur += 1
        _logger.info('generate parameters: %s' % self.cur)
        return { 'x': self.cur }

27
    def receive_trial_result(self, parameter_id, parameters, value, **kwargs):
28
        reward = extract_scalar_reward(value)
Deshui Yu's avatar
Deshui Yu committed
29
30
31
32
33
34
        _logger.info('receive trial result: %s, %s, %s' % (parameter_id, parameters, reward))
        _result.write('%d %d\n' % (parameters['x'], reward))
        _result.flush()

    def update_search_space(self, search_space):
        _logger.info('update_search_space: %s' % search_space)
QuanluZhang's avatar
QuanluZhang committed
35
        with open(os.path.join(_pwd, 'tuner_search_space.json'), 'w') as file_:
Deshui Yu's avatar
Deshui Yu committed
36
37
            json.dump(search_space, file_)

38
39
40
41
42
43
    def _on_exit(self):
        _result.close()

    def _on_error(self):
        _result.write('ERROR\n')
        _result.close()