Spaces:
Sleeping
Sleeping
File size: 2,478 Bytes
ceed47a |
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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
class BASE_C2G:
"""
This is the base class for C2G models.
Class Methods
serialize_c2g(cls, config): Serializes the configuration into a dictionary.
deserialize_c2g(self, config): Deserializes the configuration from a dictionary.
construct_model(self, configs): Abstract method for constructing a model. This needs to be implemented by subclasses.
Examples
```
>>> @register_module('test_module')
>>> class test_module(BASE_C2G):
>>> def __init__(self, a, b):
>>> self._config = {'a':a, 'b':b}
>>> self.a = a
>>> self.b = b
>>> def __call__(self):
>>> return self.a+self.b
>>> sub_module = test_module(1, 2)
>>> sub_module()
3
>>> @register_module('master_module')
>>> class master_module(BASE_C2G):
>>> def __init__(self, sub_module):
>>> self._config = {'sub_module': sub_module}
>>> self.sub_module = sub_module
>>> def __call__(self):
>>> print(self.sub_module())
>>> master_mod = master_module(sub_module)
>>> master_mod()
3
>>> save_master_module = BASE_C2G()
>>> save_master_module.serialize_c2g(master_mod._config)
{'BASE_C2G': {'SUB_MODULE': {'TEST_MODULE': {'a': 1, 'b': 2}}}}
>>> cfg._module_list
['TEST_MODULE', 'MASTER_MODULE']
```
"""
@classmethod
def serialize_c2g(cls, config):
cls_config = {}
for keyword, value in config.items():
if value.__class__.__name__.upper() not in cfg._module_list:
#if value.__name__.upper() not in cfg._module_list:
cls_config[keyword] = value
else:
cls_config[keyword.upper()] = value.serialize_c2g(value._config)
cfg._C = {cls.__name__.upper(): cls_config}
return cfg._C
def has_sub_dicts(self, config):
for key, val in config.items():
if isinstance(val, dict):
return False
return True
def deserialize_c2g(self, config):
for model, params in config.items():
pass
return cfg._dir[model](**params)
def construct_model(self, configs):
raise NotImplementedError('`construct_model` method is not implemented. To implement it, define it on the subclasses of BASE_C2G Models.')
|