Remove metrics parsing for distributed models

ertis-research · Mar 1, 2024 · 1cdb7f4 · 1cdb7f4
1 parent 975ac29
commit 1cdb7f4
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/model_training/tensorflow/mainTraining.py b/model_training/tensorflow/mainTraining.py
@@ -286,11 +286,6 @@ def create_distributed_model(self):
 
         metrics = self.metrics.replace(' ', '')
         metrics = metrics.split(',')
-        for m in metrics:
-            if m == 'tf.keras.metrics.Precision()':
-                m = tf.keras.metrics.Precision()
-            elif m == 'tf.keras.metrics.Recall()':
-                m = tf.keras.metrics.Recall()
         """Formats metrics"""
 
         outputs = []