mirror of
https://gitee.com/fastnlp/fastNLP.git
synced 2024-11-30 03:07:59 +08:00
.. | ||
__init__.py | ||
LICENSE | ||
main.py | ||
model.py | ||
README.md | ||
test.py | ||
test.txt | ||
train.py | ||
train.txt | ||
utilities.py | ||
valid.txt |
PyTorch-Character-Aware-Neural-Language-Model
This is the PyTorch implementation of character-aware neural language model proposed in this paper by Yoon Kim.
Requiredments
The code is run and tested with Python 3.5.2 and PyTorch 0.3.1.
HyperParameters
HyperParam | value |
---|---|
LSTM batch size | 20 |
LSTM sequence length | 35 |
LSTM hidden units | 300 |
epochs | 35 |
initial learning rate | 1.0 |
character embedding dimension | 15 |
Demo
Train the model with split train/valid/test data.
python train.py
The trained model will saved in cache/net.pkl
.
Test the model.
python test.py
Best result on test set: PPl=127.2163 cross entropy loss=4.8459
Acknowledgement
This implementation borrowed ideas from
https://github.com/jarfo/kchar
https://github.com/cronos123/Character-Aware-Neural-Language-Models