Skip to content
Snippets Groups Projects

clean scripts

Merged Charly Lamothe requested to merge wip_clean_scripts into master
3 files
+ 11
2
Compare changes
  • Side-by-side
  • Inline

Files

from bolsonaro.utils import save_obj_to_json, load_obj_from_json
import os
class ModelParameters(object):
def __init__(self, extracted_forest_size, normalize_D, subsets_used, normalize_weights, seed, hyperparameters):
self._extracted_forest_size = extracted_forest_size
self._normalize_D = normalize_D
self._subsets_used = subsets_used
self._normalize_weights = normalize_weights
self._seed = seed
self._hyperparameters = hyperparameters
@property
def extracted_forest_size(self):
return self._extracted_forest_size
@property
def normalize_D(self):
return self._normalize_D
@property
def subsets_used(self):
return self._subsets_used
@property
def normalize_weights(self):
return self._normalize_weights
@property
def seed(self):
return self._seed
@property
def hyperparameters(self):
return self._hyperparameters
def save(self, directory_path, experiment_id):
save_obj_to_json(directory_path + os.sep + 'model_parameters_{}.json'.format(experiment_id),
self.__dict__)
@staticmethod
def load(directory_path, experiment_id):
return load_obj_from_json(directory_path + os.sep + 'model_parameters_{}.json'.format(experiment_id),
ModelParameters)
Loading