updated labels calls in constructor and dice score plotter

utils/data_logging_utils.py

@@ -74,7 +74,7 @@ class LogWriter():
         if labels is not None:
             self.labels = self.labels_generator(labels)
         else:
-            pass
+            self.labels = ['rsfMRI']
 
         self.logger = logging.getLogger()
         file_handler = logging.FileHandler(
@@ -194,8 +194,12 @@ class LogWriter():
         ax.xaxis.set_label_position('top')
         ax.bar(np.arange(self.number_of_classes), dice_score)
         ax.set_xticks(np.arange(self.number_of_classes))
-        ax.set_xticklabels(self.labels)
-        ax.xaxis.tick_bottom()
+
+        if self.labels is None:
+            pass
+        else:
+            ax.set_xticklabels(self.labels)
+            ax.xaxis.tick_bottom()
 
         if epochs:
             self.log_writer[phase].add_figure(