【文件属性】:
文件名称:蔡氏电路matlab仿真代码-Neural-Attentive-Item-Similarity-Model:针对TKDE2018推荐的神经注意
文件大小:22.46MB
文件格式:ZIP
更新时间:2021-05-26 16:24:09
系统开源
蔡氏电路matlab仿真代码NAIS:神经注意项相似模型
这是我们对该文件的正式实现:
NAIS:推荐的神经注意项目相似性模型何湘南,何占奎,宋晶宽,刘振光,姜玉刚和蔡达生IEEE知识和数据工程学报(TKDE
2018)
两种协作过滤模型:
NAIS_concat和NAIS_prod
。
为了将模型用于隐式反馈和排名任务,我们使用对数损失和负采样对它们进行优化。
此外,我们实现了基线:
FISM
,这是众所周知的基于项目的推荐模型。
如果您使用我们的代码,请引用我们的论文。
谢谢!
通讯作者:何湘南博士()
快速上手
运行NAIS_prod:
python
NAIS.py
--dataset
pinterest-20
--pretrain
0
--weight_size
16
--embed_size
16
--data_alpha
0
--regs
[0,0,1e-6]
--alpha
0--beta
0.5
--lr
0.05
--algorithm
0
运行NAIS_concat:
python
NAIS.py
--dataset
pinterest-20
--pretrain
【文件预览】:
Neural-Attentive-Item-Similarity-Model-master
----.gitignore(12B)
----README.md(2KB)
----FISM.py(10KB)
----Data()
--------pinterest-20.train.rating(21.54MB)
--------ml-1m.test.negative(2.76MB)
--------ml-1m.test.rating(131KB)
--------ml-1m.train.rating(20.96MB)
--------pinterest-20.test.rating(842KB)
--------pinterest-20.test.negative(26.19MB)
----NAIS.py(13KB)
----Dataset.py(4KB)
----.idea()
--------workspace.xml(29KB)
----Statistics.py(724B)
----Batch_gen.py(6KB)
----Evaluate.py(3KB)
----.DS_Store(6KB)