Language Model

<style> pre { white-space: pre-wrap; } </style>

Language-Model

基於PyTorch 範例實現中文語言模型。

數據集

《三國演義》 + 《三體》

模型

RNNModel(
  (drop): Dropout(p=0.5)
  (encoder): Embedding(3719, 200)
  (rnn): GRU(200, 200, num_layers=2, dropout=0.5)
  (decoder): Linear(in_features=200, out_features=3719, bias=True)
)

生成效果

《三國演義》

《三體》