pyannote / pyannote-metrics

A toolkit for reproducible evaluation, diagnostic, and error analysis of speaker diarization systems
http://pyannote.github.io/pyannote-metrics
MIT License
183 stars 30 forks source link

fix: rewrite mapping and matching routines #31

Closed hbredin closed 5 years ago

hbredin commented 5 years ago

pyannote.core 3.x changes the return type of Annotation.mul (from xarray.DataArray to np.ndarray) because it was found that recent changes in xarray led to inconsistency in diarization metrics. "xarray" is still used in IdentificationErrorAnalysis.matrix but is meant to be removed completely in the future.

Also, "scipy" now provides an implementation of the Hungarian algorithm (supposedly faster than that of "munkres") so we switch to a recent version of "scipy".

Therefore, the code for "HungarianMapper" and "GreedyMapper" has been rewritten to rely on those two things. This also has the side effect of removing both "networkx" and "munkres" dependency.