multilingual_kws icon indicating copy to clipboard operation
multilingual_kws copied to clipboard

How to train it for more than one target_keyword?

Open twshen2000 opened this issue 2 years ago • 1 comments

Thanks!

twshen2000 avatar May 06 '22 04:05 twshen2000

I'm also looking to do this. I see there are batch methods, but the prep steps are not clear.

DCNemesis avatar Aug 18 '22 13:08 DCNemesis

Same here, I wanted to train for more than one target keywords. @mmaz Anyone figured how to do it yet? @twshen2000 @DCNemesis Thank you in advance

turinaf avatar Mar 04 '24 09:03 turinaf

Thanks for checking our work out. We did not include documentation or examples for training on multiple targets, but it's fairly straightforward to add. The main change needed is to increase the (currently hardcoded) number of target categories to be greater than 3 (which represent the categories: silence/background, unknown, and the number of target keywords). I would suggest starting with our tutorial colab and then changing transfer_learning.transfer_learn() to include CATEGORIES as a non-hardcoded argument here: https://github.com/harvard-edge/multilingual_kws/blob/5823b01f13104f45694e419aeb26a37c547914af/multilingual_kws/embedding/transfer_learning.py#L46-L53

mmaz avatar Mar 04 '24 17:03 mmaz