Skip to content

friedrichor/Language-Model-Next-Word-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Language-Model-Next-Word-Prediction

Next word prediction using language models

讲解:语言模型 实现 下一单词预测(next-word prediction)

数据集说明:

更换模型:

  1. 只需在train.py中更改model = TextRNN(n_class).to(device)即可。
  2. 代码中提供的模型为NNLM、RNNLM、RNNLM based on Attention,在model.py文件中。
    当然也可以自己编写其他语言模型,直接调用即可,更改方式同上。

About

Next word prediction using language models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages