Commit 74b6cb12 authored by Jean Ibarz's avatar Jean Ibarz
Browse files

Updated logger to write exp_config in json with indent=4 (much more pretty !)

parent 129bccef
......@@ -10,7 +10,7 @@ def experiment_logger(rootdir, logdir_prefix, exp_config, model, training_df, te
os.makedirs(name=out_dir, exist_ok=True)
model.save(filepath=os.path.join(out_dir, 'model'))
with open(os.path.join(out_dir, 'exp_config.json'), mode='wt') as exp_config_f:
json.dump(exp_config, fp=exp_config_f)
json.dump(exp_config, fp=exp_config_f, indent=4, sort_keys=True)
with open(os.path.join(out_dir, 'training_df.p'), mode='wb') as training_df_f:
pickle.dump(training_df, file=training_df_f)
with open(os.path.join(out_dir, 'test_df.p'), mode='wb') as test_df_f:
......
Supports Markdown
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