blob: dd9a358166c0d560c5ca8c24f61405a00cfb1380 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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__
|
