
GRU4REC-pytorch: PyTorch下的GRU4REC另类实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
GRU4REC-pytorch项目提供了一个不同于原始版本的PyTorch框架下GRU4REC模型的实现,便于深度学习爱好者和研究者使用。
GRU4REC-PyTorch 是 GRU4REC 模型的 PyTorch 实现版本。该代码基于 Younghun Song(YHS-968)的工作,并实现了多种损失函数,包括 TOP1、BPR、TOP1-max、BPR-max 和交叉熵损失。此实现要求使用 Python 3.5 版本和 PyTorch 0.4.1,以及 pandas 和 numpy 的相应版本。
对于数据集的准备,请注意 RecSys Challenge 2015 数据集可以从相应的资源获取并进行预处理以获得训练数据和测试数据。在实际应用中,仅使用训练集合,并且不考虑原始提供的测试集。具体来说,在这个实现里,会将输入的数据进一步划分为用于模型训练的子集以及用于验证性能的子集(即最后一天的数据)。
文件格式应遵循 RecSys Challenge 2015 的规范:
- 训练数据应当命名为 recSys15TrainOnly.txt。
- 测试数据则需要保存为 recSys15Valid.txt。
全部评论 (0)
还没有任何评论哟~


