
caser_pytorch:基于PyTorch的卷积序列嵌入推荐模型(Caser)源码
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Caser_pytorch项目提供了一个使用PyTorch实现的Caser(Convolutional Sequence Embedding Recommendation Model)推荐系统模型,适用于序列数据预测。该项目包含了详细的文档和示例代码,适合研究与开发人员参考学习。
卡塞尔·皮尔·火炬(Caser-PyTorch)是卷积序列嵌入推荐模型(Caser)的PyTorch实现:通过卷积序列嵌入进行个性化Top-N序列推荐,出自唐佳西和王珂在WSDM 18上的论文。该实现支持Python 2或3版本。
安装所需的软件包时,请遵循暴躁科学用法中的指示。
运行命令`python train_caser.py`以开始训练模型。配置数据集需分为两个单独的文件:train.txt 和 test.txt 。每个文件包含三元组集合,即用户、项目和评分信息;这些三元组按照时间顺序排列。由于这是一个序列推荐问题,因此评级信息无关紧要,并被转换为1。
在`train_caser.py`中定义模型参数:
- L :序列长度
- T :目标数量
- d :潜在维度数
- nv :垂直滤波器的数量
- nh :水平滤波器的数量
- ac_conv:卷积层的激活函数(如论文中的phi_c)
- ac_fc:全连接层的激活函数
全部评论 (0)
还没有任何评论哟~


