search

wq2012 / SpectralCluster

Python re-implementation of the (constrained) spectral clustering algorithms used in Google's speaker diarization papers.
https://google.github.io/speaker-id/publications/LstmDiarization/
Apache License 2.0
513 stars 73 forks source link

Constrained MultiStageClusterer #49

Closed niemiaszek closed 1 year ago

niemiaszek commented 1 year ago

Is there a chance one could use speaker turns constraints with MultiStageClusterer or is the lack of support for this intended? I see that streaming_predict method is lacking constraint_matrix argument. Adding it and calling self.main.predict with passed constraint_matrix seems to be the solution for this issue.

wq2012 commented 1 year ago

Currently we are not doing constraint for multi-stage starting at num_embeddings>=U1.

It's doable for AHC but a bit too complicated compared with the benefits there.