NeuMF 的 gluon 实现

  Posted by Mr.Zhangon 30 Jul, 2019

   MXNET   

Xiangnan He et al.,2017,Neural Collaborative Filtering(http://dl.acm.org/citation.cfm?id=3052569)论文的gluon复现,原作的代码github(https://github.com/hexiangnan/neuralcollaborativefiltering/)。NeuMF模型分为两块:GMF与MLP,具体架构如下图: NeuMF(/static/upload/201907/NeuMF.png) 本文作二分类问题,user与item分别embedding后丢入GMF作内积;丢入MLP学[...]

mxnet 混合编程

  Posted by Mr.Zhangon 24 Jun, 2019

   MXNET    PYTHON   

深度学习几个python库,可以分为命令式编程和符号式编程,前者以PyTorch为代表,后者主要是Tensorflow,符号式编程不易理解与调试,所以Keras成为大多数人的选择,目前官方支持中文文档。除此之外,还有一种选择,那就是mxnet,支持两类模式的混合编程,很好很强大。 在设计模型时候,可以符号式方法丢数据进去看看中间量;正式跑参数的时候可以调用hybridize函数加速性能,减少等待时间,十分适合炼丹。官方介绍地址是: , 感觉太过简略。 一个简单模型设计: python class MLP(gluon.HybridBlock): 使用混[...]