"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "9f4c4f5e8208d4c54eb24814750753f1e005d854"
Commit 1362de02 authored by Yan Ni's avatar Yan Ni Committed by QuanluZhang
Browse files

fix dev-csv (#966)

add trial id to csv and remove pandas
parent 3beaa3fd
...@@ -50,7 +50,6 @@ setup( ...@@ -50,7 +50,6 @@ setup(
'hyperopt', 'hyperopt',
'json_tricks', 'json_tricks',
'numpy', 'numpy',
'pandas',
'psutil', 'psutil',
'pyyaml', 'pyyaml',
'requests', 'requests',
......
...@@ -478,6 +478,7 @@ def parse_trial_data(content): ...@@ -478,6 +478,7 @@ def parse_trial_data(content):
for trial_data in content: for trial_data in content:
for phase_i in range(len(trial_data['hyperParameters'])): for phase_i in range(len(trial_data['hyperParameters'])):
hparam = json.loads(trial_data['hyperParameters'][phase_i])['parameters'] hparam = json.loads(trial_data['hyperParameters'][phase_i])['parameters']
hparam['id'] = trial_data['id']
if 'finalMetricData' in trial_data.keys() and phase_i < len(trial_data['finalMetricData']): if 'finalMetricData' in trial_data.keys() and phase_i < len(trial_data['finalMetricData']):
reward = json.loads(trial_data['finalMetricData'][phase_i]['data']) reward = json.loads(trial_data['finalMetricData'][phase_i]['data'])
if isinstance(reward, (float, int)): if isinstance(reward, (float, int)):
...@@ -493,7 +494,6 @@ def parse_trial_data(content): ...@@ -493,7 +494,6 @@ def parse_trial_data(content):
def export_trials_data(args): def export_trials_data(args):
"""export experiment metadata to csv """export experiment metadata to csv
:param: csv file path to export
""" """
nni_config = Config(get_config_filename(args)) nni_config = Config(get_config_filename(args))
rest_port = nni_config.get_config('restServerPort') rest_port = nni_config.get_config('restServerPort')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment