New Model running

utils/project_settings.py

@@ -1,7 +1,5 @@
 from argparse import Namespace
 
-from ml_lib.utils.config import Config
-
 
 class DataClass(Namespace):
 
@@ -18,18 +16,19 @@ class DataClass(Namespace):
         return f'{self.__class__.__name__}({self.__dict__().__repr__()})'
 
     def __getitem__(self, item):
-        return self.__getattribute__(item)
+        return self.__dict__()[item]
 
 
 class Classes(DataClass):
+
     # Object Classes for Point Segmentation
     Sphere = 0
     Cylinder = 1
     Cone = 2
-    Box = 3
-    Polytope = 4
+    Box = 3         # All SubTypes of Planes
+    Polytope = 4    #
     Torus = 5
-    Plane = 6
+    Plane = 6       #
 
 
 class Settings(DataClass):
@@ -38,6 +37,11 @@ class Settings(DataClass):
     PN2 = 'pc'
 
 
+class ClusterTypes(DataClass):
+    prim = 'prim'
+    grid = 'grid'
+    none = ''
+
 class DataSplit(DataClass):
     # DATA SPLIT OPTIONS
     train = 'train'
@@ -59,4 +63,4 @@ class GlobalVar(DataClass):
 
     prim_count = -1
 
-    settings = Settings()
+    settings = Settings()