From 48f4d88bffd1be66e0b84d1aa3ce3c86880d66d5 Mon Sep 17 00:00:00 2001 From: Kibrewossen <47031138+kebtes@users.noreply.github.com> Date: Sat, 17 May 2025 13:33:19 +0300 Subject: [PATCH 1/2] Rename LAYER_CLASSES to LAYER_REGISTERY --- nnf/models/model.py | 2 +- nnf/utils/__init__.py | 4 ++-- nnf/utils/layer_config.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/nnf/models/model.py b/nnf/models/model.py index dd629eb..ee165db 100644 --- a/nnf/models/model.py +++ b/nnf/models/model.py @@ -11,7 +11,7 @@ from nnf.layers.base import Layer from nnf.losses.base import Loss from nnf.optimizers.base import Optimizer -from nnf.utils import LAYER_CLASSES +from nnf.utils import LAYER_REGISTERY """ This module defines a neural network Model class that combines layers and diff --git a/nnf/utils/__init__.py b/nnf/utils/__init__.py index c9ab817..ea75386 100644 --- a/nnf/utils/__init__.py +++ b/nnf/utils/__init__.py @@ -1,3 +1,3 @@ -from nnf.utils.layer_config import LAYER_CLASSES +from nnf.utils.layer_config import LAYER_REGISTERY -__all__ = ['LAYER_CLASSES'] \ No newline at end of file +__all__ = ['LAYER_REGISTERY'] \ No newline at end of file diff --git a/nnf/utils/layer_config.py b/nnf/utils/layer_config.py index fafbf5a..5dd86e9 100644 --- a/nnf/utils/layer_config.py +++ b/nnf/utils/layer_config.py @@ -13,7 +13,7 @@ from nnf.optimizers import GradientDescent from nnf.optimizers import Momentum -LAYER_CLASSES = { +LAYER_REGISTERY = { "Dense" : Dense, "ReLU" : ReLU, "LeakyRelU" : LeakyReLU, From 2739f42c156cd2aba3eb86d7f2aaa4876d3926c3 Mon Sep 17 00:00:00 2001 From: Kibrewossen <47031138+kebtes@users.noreply.github.com> Date: Sat, 17 May 2025 13:33:59 +0300 Subject: [PATCH 2/2] Update references from LAYER_CLASSES to LAYER_REGISTERY in model attribute management --- nnf/models/model.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/nnf/models/model.py b/nnf/models/model.py index ee165db..e309e1e 100644 --- a/nnf/models/model.py +++ b/nnf/models/model.py @@ -401,16 +401,16 @@ def set_model_attrs(self, attrs : Dict): Args: attrs (dict): A dictionary of attribute names and values to set. If a value is a dictionary with keys "type" and "attrs", - it initializes a layer object from `LAYER_CLASSES` and sets its parameters. + it initializes a layer object from `LAYER_REGISTERY` and sets its parameters. """ for key, val in attrs.items(): if isinstance(val, dict): - obj = LAYER_CLASSES[val["type"]]() + obj = LAYER_REGISTERY[val["type"]]() obj.set_params(val["attrs"]) setattr(self, key, obj) - elif isinstance(val, str) and val in LAYER_CLASSES: - setattr(self, key, LAYER_CLASSES[val]()) + elif isinstance(val, str) and val in LAYER_REGISTERY: + setattr(self, key, LAYER_REGISTERY[val]()) else: setattr(self, key, val) @@ -508,13 +508,13 @@ def load_model(file_path : str): attrs = layer["attrs"] if layer["attrs"] else {} - obj : Layer = LAYER_CLASSES[layer_type]() + obj : Layer = LAYER_REGISTERY[layer_type]() obj.set_params(attrs) layers.append(obj) model_attrs = { - key: LAYER_CLASSES[val]() if isinstance(val, str) and val in LAYER_CLASSES else val + key: LAYER_REGISTERY[val]() if isinstance(val, str) and val in LAYER_REGISTERY else val for key, val in model_attrs.items() }