lm_word_level:语言模型单词级别 源码
词级语言建模RNN 本示例在语言建模任务上训练多层RNN(Elman,GRU或LSTM)。 默认情况下,训练脚本使用提供的Wikitext-2数据集。 然后,生成的脚本可以使用经过训练的模型来生成新文本。 python main.py --cuda --epochs 6 # Train a LSTM on Wikitext-2 with CUDA python main.py --cuda --epochs 6 --tied # Train a tied LSTM on Wikitext-2 with CUDA python main.py --cuda --ep
文件列表
lm_word_level-master.zip
(预估有个10文件)
lm_word_level-master
main.py
10KB
data
wikitext-2
valid.txt
1.07MB
test.txt
1.2MB
train.txt
10.3MB
README
159B
generate.py
3KB
requirements.txt
6B
暂无评论