inversion json files

1 files changed, 27 insertions, 0 deletions

diff --git a/inversion/params.py b/inversion/params.py
new file mode 100644
index 0000000..dd9a358
--- /dev/null
+++ b/inversion/params.py
@@ -0,0 +1,27 @@
+# ------------------------------------------------------------------------------
+# Util class for hyperparams.
+# ------------------------------------------------------------------------------
+
+import json
+
+class Params():
+  """Class that loads hyperparameters from a json file."""
+
+  def __init__(self, json_path):
+    self.update(json_path)
+
+  def save(self, json_path):
+    """Saves parameters to json file."""
+    with open(json_path, 'w') as f:
+      json.dump(self.__dict__, f, indent=4)
+
+  def update(self, json_path):
+    """Loads parameters from json file."""
+    with open(json_path) as f:
+      params = json.load(f)
+      self.__dict__.update(params)
+
+  @property
+  def dict(self):
+    """Gives dict-like access to Params instance."""
+    return self.__dict__