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

Add a `partial_fit()` API #19

Open wq2012 opened 5 years ago

wq2012 commented 5 years ago

Add a new API named partial_fit(), which allows the user to train on a single sequence.

This API will target at the advanced users. The user needs to write his own code for things like data shuffling.

Ideally, the fit() function should be calling the partial_fit() function.