From fd0c42c77a44f09ad76dda39132c546695b5ab3d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20Bredin?= Date: Fri, 28 Jan 2022 14:08:17 +0100 Subject: [PATCH] fix: fix computation of same_chunk matrix --- pyannote/audio/pipelines/speaker_diarization.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pyannote/audio/pipelines/speaker_diarization.py b/pyannote/audio/pipelines/speaker_diarization.py index da4d0bb58..89ab11c6e 100644 --- a/pyannote/audio/pipelines/speaker_diarization.py +++ b/pyannote/audio/pipelines/speaker_diarization.py @@ -417,7 +417,10 @@ def oracle_cost_func(Y, y): for i in permutation ] ) - chunk_idx.extend([c] * len(permutation)) + chunk_idx.extend([ + c if ((i is not None) and (active_reference[i])) else -1 + for i in permutation + ]) oracle_clusters = np.array(oracle_clusters) oracle = 1.0 * squareform(pdist(oracle_clusters, metric="equal"))