def _save(self, path):
"""Save the ErrorAnalysisManager to the given path.
:param path: The directory path to save the ErrorAnalysisManager to.
:type path: str
"""
top_dir = Path(path)
top_dir.mkdir(parents=True, exist_ok=True)
if len(self._ea_config_list) != len(self._ea_report_list):
raise ConfigAndResultMismatchException(
"The number of error analysis configs {0} doesn't match the "
"number of results {1}".format(len(self._ea_config_list),
len(self._ea_report_list)))
for index in range(0, len(self._ea_report_list)):
# save the configs
directory_manager = DirectoryManager(parent_directory_path=path)
config_path = (directory_manager.create_config_directory() /
'config.json')
ea_config = self._ea_config_list[index]
with open(config_path, 'w') as file:
json.dump(ea_config, file, default=config_json_converter)
# save the reports
report_path = (directory_manager.create_data_directory() /
'report.json')
ea_report = self._ea_report_list[index]
with open(report_path, 'w') as file:
json.dump(ea_report, file, default=report_json_converter)
def _save(self, path):
"""
Save the DataBalanceManager to the given path.
:param path: The directory path to save the DataBalanceManager to.
:type path: str
"""
top_dir = Path(path)
top_dir.mkdir(parents=True, exist_ok=True)
# If manager is not added to RAIInsights, don't save any of its configs
if not self._is_added:
return
dir_manager = DirectoryManager(parent_directory_path=path)
config_dir = dir_manager.create_config_directory()
manager_path = config_dir / MANAGER_JSON
with open(manager_path, "w") as f:
json.dump(self.list(), f)
# if measures have been computed, save the computed measures
if self._data_balance_measures:
measures_path = config_dir / MEASURES_JSON
with open(measures_path, "w") as f:
json.dump(self._data_balance_measures, f)
data_path = dir_manager.create_data_directory() / DATA_JSON
self._df.to_json(data_path, orient="split")
def _save(self, path):
"""Save the CausalManager to the given path.
:param path: The directory path to save the CausalManager to.
:type path: str
"""
causal_dir = Path(path)
causal_dir.mkdir(parents=True, exist_ok=True)
# Save results to disk
for result in self._results:
directory_manager = DirectoryManager(parent_directory_path=path)
data_path = directory_manager.create_data_directory()
result.save(data_path)
def _save(self, path):
"""Save the CounterfactualManager to the given path.
:param path: The directory path to save the CounterfactualManager to.
:type path: str
"""
counterfactual_dir = Path(path)
counterfactual_dir.mkdir(parents=True, exist_ok=True)
for counterfactual_config in self._counterfactual_config_list:
directory_manager = DirectoryManager(parent_directory_path=path)
counterfactual_config.save_config(
directory_manager.create_config_directory())
counterfactual_config.save_result(
directory_manager.create_data_directory())
counterfactual_config.save_explainer(
directory_manager.create_generators_directory())
def _save(self, path):
"""Save the ExplainerManager to the given path.
:param path: The directory path to save the ExplainerManager to.
:type path: str
"""
top_dir = Path(path)
top_dir.mkdir(parents=True, exist_ok=True)
if self._is_added:
directory_manager = DirectoryManager(parent_directory_path=path)
data_directory = directory_manager.create_data_directory()
# save the explanation
if self._explanation:
save_explanation(self._explanation,
data_directory / ManagerNames.EXPLAINER)
meta = {IS_RUN: self._is_run, IS_ADDED: self._is_added}
with open(data_directory / META_JSON, 'w') as file:
json.dump(meta, file)