Skip to content
Snippets Groups Projects

Luc manage normalization

Merged Luc Giffon requested to merge luc_manage_normalization into wip_clean_scripts
3 files
+ 30
3
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -4,10 +4,11 @@ import os
class ModelParameters(object):
def __init__(self, forest_size, extracted_forest_size, seed=None):
def __init__(self, forest_size, extracted_forest_size, normalize, seed=None):
self._forest_size = forest_size
self._extracted_forest_size = extracted_forest_size
self._seed = seed
self._normalize = normalize
@property
def forest_size(self):
@@ -21,12 +22,17 @@ class ModelParameters(object):
def seed(self):
return self._seed
@property
def normalize(self):
return self._normalize
def save(self, directory_path, experiment_id):
with open(directory_path + os.sep + 'model_parameters_{}.json'.format(experiment_id), 'w') as output_file:
json.dump({
'forest_size': self._forest_size,
'extracted_forest_size': self._extracted_forest_size,
'seed': self._seed
'seed': self._seed,
'normalize': self._normalize
},
output_file,
indent=4)
Loading