sea-wyq/NRMS-Pytorch

基于pytorch实现NRMS

#Pytorch-NRMS实现 #

数据集准备

• MIND新闻推荐数据集。
• Glove 预训练词向量。

数据处理

   ProcessMindByGlove.py

训练模型

   train.py