【文件属性】:
文件名称:蔡氏电路matlab仿真代码-nfm:nfm
文件大小:2.97MB
文件格式:ZIP
更新时间:2021-05-26 16:33:50
系统开源
蔡氏电路matlab仿真代码神经分解机
这是我们对本文的实现:
何湘南和蔡达生(2017)。
[用于稀疏预测分析的神经分解机。]()在SIGIR
'17的会议记录中,新宿,东京,日本,2017年8月7日至11日。
在我们建议的神经网络框架下,我们还发布了分解机的TensorFlow实现。
如果您使用我们的代码,请引用我们的SIGIR'17论文。
谢谢!
作者:何湘南博士()
运行代码的示例。
python
NeuralFM.py
--dataset
frappe
--hidden_factor
64
--layers
[64]
--keep_prob
[0.8,0.5]
--loss_type
square_loss
--activation
relu
--pretrain
0
--optimizer
AdagradOptimizer
--lr
0.05
--batch_norm
1
--verbose
1
--early_stop
1
--epoch
200
在代码中已经清楚地说明了命令的指令(请参阅parse_args函数)。
当前的实现支持两项任务:回归和二进制分类。
回归任务优
【文件预览】:
nfm-master
----LoadData.py(4KB)
----.gitignore(6B)
----data()
--------frappe()
----__init__.py(0B)
----NeuralFM.py(19KB)
----README.md(1KB)
----FM.py(16KB)
----1.py(772B)