<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)
)

生成效果

《三国演义》

《三体》