search

ml5js / ml5-data-and-models

Data sets and pre-trained models for ml5.js
https://ml5js.org/docs/data-overview
MIT License
125 stars 98 forks source link

Update LSTM training with keras #12

Closed cvalenzuela closed 6 years ago

cvalenzuela commented 6 years ago

The training example needs to be updated to match the new script we are using. (https://github.com/keras-team/keras/blob/master/examples/lstm_text_generation.py) I'll push some changes to this by the end of the day

shiffman commented 6 years ago

Thank you @cvalenzuela. Tagging @nikitahuggins so she knows to follow too!

shiffman commented 6 years ago

@cvalenzuela should I remove this from the June 15th beta and we'll tackle later if/when performance improves with the latest tfjs LSTM models?

cvalenzuela commented 6 years ago

Yes, this should be removed because we have reverted to the previous implementation of the LSTM. See https://github.com/ml5js/ml5-library/pull/150. I'm currently working on creating a tutorial with the script we already have

cvalenzuela commented 6 years ago

closing this because we ended up reverting this class.