google / uis-rnn

This is the library for the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm, corresponding to the paper Fully Supervised Speaker Diarization.
https://arxiv.org/abs/1810.04719
Apache License 2.0
1.55k stars 320 forks source link

Consider allowing enforcing max number of speakers in predict() #26

Open wq2012 opened 5 years ago

wq2012 commented 5 years ago

Although the whole idea of UIS-RNN is to handle unbounded number of speakers by learning from data, some people may still want to enforce the max number of speakers in predict.

Consider adding this option in the future.