Skip to content
Snippets Groups Projects

clean scripts

Merged Charly Lamothe requested to merge wip_clean_scripts into master
Compare and Show latest version
12 files
+ 99
307
Compare changes
  • Side-by-side
  • Inline

Files

import json
from bolsonaro.utils import save_obj_to_json, load_obj_from_json
import os
@@ -37,27 +38,10 @@ class DatasetParameters(object):
return self._dataset_normalizer
def save(self, directory_path, experiment_id):
with open(directory_path + os.sep + 'dataset_parameters_{}.json'.format(experiment_id), 'w') as output_file:
json.dump({
'name': self._name,
'test_size': self._test_size,
'dev_size': self._dev_size,
'random_state': self._random_state,
'dataset_normalizer': self._dataset_normalizer,
'train_on_subset': self._train_on_subset
},
output_file,
indent=4)
save_obj_to_json(directory_path + os.sep + 'dataset_parameters_{}.json'.format(experiment_id),
self.__dict__)
@staticmethod
def load(directory_path, experiment_id):
with open(directory_path + os.sep + 'dataset_parameters_{}.json'.format(experiment_id), 'r') as input_file:
parameters = json.load(input_file)
return DatasetParameters(
name=parameters['name'],
test_size=parameters['test_size'],
dev_size=parameters['dev_size'],
random_state=parameters['random_state'],
dataset_normalizer=parameters['dataset_normalizer'],
train_on_subset=parameters['train_on_subset']
)
return load_obj_from_json(directory_path + os.sep + 'dataset_parameters_{}.json'.format(experiment_id),
DatasetParameters)
Loading