Skip to content

Commit

Permalink
Added Markdown formatting to examples/imdb_lstm.py (keras-team#12313)
Browse files Browse the repository at this point in the history
* Added MarkDown formatting to examples/imdb_fasttext.py

* Added MarkDown formatting to examples/imdb_lstm.py
  • Loading branch information
joosephook authored and gabrieldemarmiesse committed Feb 21, 2019
1 parent a7fdfdf commit 304b395
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions examples/imdb_lstm.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,18 @@
'''Trains an LSTM model on the IMDB sentiment classification task.
'''
#Trains an LSTM model on the IMDB sentiment classification task.
The dataset is actually too small for LSTM to be of any advantage
compared to simpler, much faster methods such as TF-IDF + LogReg.
# Notes
**Notes**
- RNNs are tricky. Choice of batch size is important,
choice of loss and optimizer is critical, etc.
Some configurations won't converge.
- LSTM loss decrease patterns during training can be quite different
from what you see with CNNs/MLPs/etc.
'''
from __future__ import print_function

Expand Down

0 comments on commit 304b395

Please sign in to comment.