Skip to content

SpringtoString/BPR-Torch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BPR-Torch

BPR算法是基础的推荐算法,在我入门推荐系统时很容易理解BPR算法,但是当我想实现完整的推荐算法时,却困难重重,如何加载数据,如何进行负采样,如何写评价指标函数,让我很烦恼。 参照王翔老师的NGCF算法 Tensorflow版PyTorch版 自己实现一个完整的BPR推荐算法框架。

我对此进行了一些改进,想把整个代码封装成一个BPR类,但是由于cuda和并行评价的冲突,导致BPR类代码在类的思想有点瑕疵,还需再改进

  • 运行方法 python main.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages