ml5js / training-charRNN

Training charRNN model for ml5js
Other
96 stars 46 forks source link

Training on word level #4

Open vvzen opened 5 years ago

vvzen commented 5 years ago

First of all: thanks for doing this! It's an awesome achievement in terms of making these tools accessible to everybody. 🎨🖌

I'm trying to create a model based on a series of poems, is there a way to train it on words? I would like to compare the results of training the rnn on char sequences or word sequences.

Thanks!

cvalenzuela commented 5 years ago

This example is based on https://github.com/sherjilozair/char-rnn-tensorflow, a character level model.

Perhaps parsing the generated result and grabbing the words might work?

shiffman commented 5 years ago

We could also look at porting something like this and making a flag in train.py?

https://github.com/hunkim/word-rnn-tensorflow