【文件属性】：
文件名称：蔡氏方程matlab求解代码-neural_factorization_machine:神经分解机的TenforFlow实现
文件大小：20.04MB
文件格式：ZIP
更新时间：2021-05-19 21:44:41
系统开源
蔡氏方程matlab重构代码神经分解机
这是我们对本文的实现：
何湘南和蔡达生（2017）。
2017年8月7日至11日，日本东京新宿，SIGIR
'17会议论文集。
在我们建议的神经网络框架下，我们还发布了分解机的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函数）。
当前的实现支持两项任务：回归和二进制分类。
回归任务优化了RMSE，而二进制分类任务优化了Log
L
【文件预览】：
neural_factorization_machine-master
----README.md(1KB)
----LoadData.py(4KB)
----FM.py(16KB)
----NeuralFM.py(19KB)
----data()
--------ml-tag()
--------frappe()