-
使用PyTorch训练大规模分解模型以实现协同过滤的recoder
资源介绍
编码器
介绍
Recoder是一种快速实现，用于在最近的工作之后使用基于小批量的负采样来训练协作过滤潜在因子模型：
。
Recoder包含因式分解模型的两种实现：自动编码器和矩阵因式分解。
查看和。
安装
建议使用python 3.8。 不支持Python 2。
pip install -U recsys-recoder
例子
请查看scripts/目录，以获取有关不同数据集的一些良好示例。 您可以在Nvidia Tesla K80 GPU上不到一分钟的时间内，获得均方误差完全训练的MovieLens-20M数据集。
进一步阅读
引用
如果对您的研究有所帮助，请在您的出版物中引用该论文：
@inproceedings{recoder,
author = {Moussawi, Abdallah},
title = {Towards Large Scale Training