File size: 1,319 Bytes
3494c6b |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import json
from dataclasses import dataclass
from dataclasses import asdict
@dataclass
class HyperParams:
"""
Simple wrapper to store hyperparameters for Python-based rewriting methods.
"""
@classmethod
def from_json(cls, fpath):
with open(fpath, "r") as f:
data = json.load(f)
return cls(**data)
def construct_float_from_scientific_notation(config: dict):
for key, value in config.items():
if isinstance(value, str):
try:
# Convert scalar to float if it is in scientific notation format
config[key] = float(value)
except:
pass
return config
def to_dict(config) -> dict:
dict = asdict(config)
return dict
# @classmethod
# def from_hparams(cls, hparams_name_or_path: str):
#
# if '.yaml' not in hparams_name_or_path:
# hparams_name_or_path = hparams_name_or_path + '.yaml'
# config = compose(hparams_name_or_path)
#
# assert config.alg_name in ALG_DICT.keys() or print(f'Editing Alg name {config.alg_name} not supported yet.')
#
# params_class, apply_algo = ALG_DICT[config.alg_name]
#
# return params_class(**config)
|