Update speechscore.py

speechscore.py

@@ -20,7 +20,7 @@ from scores.cbak import CBAK
 from scores.covl import COVL
 from scores.mcd import MCD
 
-def compute_mean_results(*results):
+def compute_mean_results(*results, round_digits=None):
     mean_result = {}
 
     # Use the first dictionary as a reference for keys
@@ -28,10 +28,13 @@ def compute_mean_results(*results):
         # If the value is a nested dictionary, recurse
         if isinstance(results[0][key], dict):
             nested_results = [d[key] for d in results]
-            mean_result[key] = compute_mean_results(*nested_results)
+            mean_result[key] = compute_mean_results(*nested_results, round_digits=round_digits)
         # Otherwise, compute the mean of the values
         else:
-            mean_result[key] = sum(d[key] for d in results) / len(results)
+            if round_digits is not None:
+                mean_result[key] = round(sum(d[key] for d in results) / len(results), round_digits)
+            else:
+                mean_result[key] = sum(d[key] for d in results) / len(results)
 
     return mean_result
 
@@ -46,7 +49,7 @@ class ScoresList:
     def __str__(self):
         return 'Scores: ' + ' '.join([x.name for x in self.scores])
 
-    def __call__(self, test_path, reference_path, window=None, score_rate=None, return_mean=False):
+    def __call__(self, test_path, reference_path, window=None, score_rate=None, return_mean=False, round_digits=None):
         """
         window: float
             the window length in seconds to use for scoring the files.
@@ -68,7 +71,7 @@ class ScoresList:
             data['rate'] = score_rate
             
             for score in self.scores:
-                result_score = score.scoring(data, window, score_rate)
+                result_score = score.scoring(data, window, score_rate, round_digits)
                 results[score.name] = result_score
         else:    
             if os.path.isdir(test_path):
@@ -81,18 +84,18 @@ class ScoresList:
                     else:
                         data = self.audio_reader(test_path+'/'+audio_id, None)
                     for score in self.scores:
-                        result_score = score.scoring(data, window, score_rate)
+                        result_score = score.scoring(data, window, score_rate, round_digits)
                         results_id[score.name] = result_score
                     results[audio_id] = results_id
                 
             elif os.path.isfile(test_path):           
                 data = self.audio_reader(test_path, reference_path)
                 for score in self.scores:
-                    result_score = score.scoring(data, window, score_rate)
+                    result_score = score.scoring(data, window, score_rate, round_digits)
                     results[score.name] = result_score
                 
         if return_mean:
-            mean_result = compute_mean_results(*results.values())
+            mean_result = compute_mean_results(*results.values(), round_digits=round_digits)
             results['Mean_Score'] = mean_result
 
         return results