[Error Update]:Correcting the DER scoring.

s4d/scoring.py

@@ -25,22 +25,22 @@ class DER:
     :attr not_assigned: a list of hypothesis cluster_list not associated to a reference cluster
 
     """
-    def __init__(self, hyp_diarization, ref_diarization, uem_dirization = None, collar=0, no_overlap=False):
+    def __init__(self, hyp_diarization, ref_diarization, uem_diarization = None, collar=0, no_overlap=False):
         """
 
         :param hyp_diarization: a hypothesis Diar object
         :param ref_diarization: a reference Diar object
-        :param uem_dirization: a uem Diar object
+        :param uem_diarization: a uem Diar object
         :param collar: is the no-score zone around reference speaker segments
         :param no_overlap: to limit scoring to those time regions in which only
         a single speaker is speaking
 
         """
-        uem = uem_dirization
+        uem = uem_diarization
         self.collar = collar
         self.match = None
-        if uem_dirization is not None:
-            uem = copy.deepcopy(uem_dirization)
+        if uem_diarization is not None:
+            uem = copy.deepcopy(uem_diarization)
             if len(uem.unique('cluster')) > 1:
                 raise Exception('UEM contains more than one cluster')
             if len(uem.unique('show')) > 1:
@@ -54,15 +54,15 @@ class DER:
             raise Exception('REF contains more than one show')
         else:
             self.show = lst[0]
-        if uem_dirization is None:
+        if uem_diarization is None:
             print('uem from ref')
-            self.uem_set = set([i for i in range(ref[0]['start'], ref[-1]['stop']+1)])
+            self.uem_set = set([i for i in range(min(ref.unique('start')), max(ref.unique('stop'))+1)])
 
         if collar > 0:
             rem = list()
             for row in ref:
-                rem += [i for i in range(row['start']-collar, row['start']+collar+1)]
-                rem += [i for i in range(row['stop']-collar, row['stop']+collar+1)]
+                rem += [i for i in range(row['start']-collar, row['start']+collar)]
+                rem += [i for i in range(row['stop']-collar, row['stop']+collar)]
             self.uem_set_collar = self.uem_set - set(rem)
         else:
             self.uem_set_collar = self.uem_set