diff --git a/model/src/utils/__init__.py b/model/src/utils/__init__.py
index 54c4ddf..56ac462 100644
--- a/model/src/utils/__init__.py
+++ b/model/src/utils/__init__.py
@@ -1,2 +1,3 @@
+from .get_class import get_class
 from .load_model import DEVICE, load_model
 from .vcda_dataset import VCDADataset
diff --git a/model/src/utils/get_class.py b/model/src/utils/get_class.py
new file mode 100644
index 0000000..3b245b0
--- /dev/null
+++ b/model/src/utils/get_class.py
@@ -0,0 +1,10 @@
+from importlib import import_module
+from types import ModuleType
+
+
+def get_class(path: str) -> ModuleType:
+    parts = path.split('.')
+    module_path = '.'.join(parts[:-1])
+    class_name = parts[-1]
+    module = import_module(module_path)
+    return getattr(module, class_name)