蔡氏电路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函数)。
当前的实现支持两项任务:回归和二进制分类。
回归任务优
2022-03-15 08:42:30
2.97MB
系统开源
1