feat: add save_dict_into_h5();

pypots/data/__init__.py

@@ -22,6 +22,7 @@
  pickle_load,
  pickle_dump,
 )
+from .saving import save_dict_into_h5
 
 __all__ = [
  # datasets
@@ -39,4 +40,6 @@
  "mcar",
  "pickle_load",
  "pickle_dump",
+ # saving
+ "save_dict_into_h5",
 ]

pypots/data/saving.py

@@ -0,0 +1,43 @@
+"""
+Data saving utilities.
+"""
+
+# Created by Wenjie Du <[email protected]>
+# License: GLP-v3
+
+
+import os
+
+import h5py
+
+from pypots.utils.file import create_dir_if_not_exist
+from pypots.utils.logging import logger
+
+
+def save_dict_into_h5(data_dict: dict, saving_dir: str) -> None:
+ """Save the given data (in a dictionary) into the given h5 file.
+
+ Parameters
+ ----------
+ data_dict : dict,
+ The data to be saved, should be a Python dictionary.
+
+ saving_dir : str,
+ The h5 file to save the data.
+
+ """
+
+ def save_set(handle, name, data):
+ if isinstance(data, dict):
+ single_set_handle = handle.create_group(name)
+ for key, value in data.items():
+ save_set(single_set_handle, key, value)
+ else:
+ handle.create_dataset(name, data=data)
+
+ create_dir_if_not_exist(saving_dir)
+ saving_path = os.path.join(saving_dir, "datasets.h5")
+ with h5py.File(saving_path, "w") as hf:
+ for k, v in data_dict.items():
+ save_set(hf, k, v)
+ logger.info(f"Successfully saved the given data into {saving_path}.")