Better fit() API, to accept list of sequences

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.56k stars 319 forks source link

Better fit() API, to accept list of sequences #22

Closed wq2012 closed 5 years ago

wq2012 commented 5 years ago

fit() should accept list of arrays for both training sequences and training cluster IDs, and do the concatenation for the user.

The current API requires the user to do the concatenation himself/herself, which could cause unnecessary confusions.