ml5js / ml5-library

Friendly machine learning for the web! 🤖
https://ml5js.org
Other
6.45k stars 905 forks source link

Add word-rnn #543

Open noisyneuron opened 5 years ago

noisyneuron commented 5 years ago

→ Description 📝

New Feature: Add the word-rnn model -- its similar to char-rnn but trains on a word level instead -- producing interesting results with less time / compute. There is a tensorflow implementation of this already -- models could be converted to tfjs. Suggestions on datasets to use would be great! Yelp reviews? Wikipedia text?

joeyklee commented 5 years ago

@noisyneuron - Very cool! Regarding datasets, it might be interesting to reach out to @nikitahuggins regarding her work with Afro-American authors and writers. Just an idea.

noisyneuron commented 5 years ago

@joeyklee that's a great idea! Will get to the datasets once I have the core technical stuff figured