added labels generator function

utils/data_logging_utils.py

@@ -19,6 +19,7 @@ import shutil
 import logging
 import numpy as np
 import re
+from textwrap import wrap
 
 # The SummaryWriter class provides a high-level API to create an event file in a given directory and add summaries and events to it.
 # More here: https://tensorboardx.readthedocs.io/en/latest/tensorboard.html
@@ -92,8 +93,16 @@ class LogWriter():
         Raises:
             None
         """
-    
-        return pass
+
+        label_classes = []
+
+        for label in labels:
+
+            label_class = re.sub(r'([a-z](?=[A-Z])|[A-Z](?=[A-Z][a-z]))', r'\1 ', label)
+            label_class = ['\n'.join(wrap(element, 40)) for element in label_class]
+            label_classes.append(label_class)
+
+        return label_classes
 
     def log(self, message):
         """Log function