推荐系统BPRMF模型的Pytorch实现,基于MovieLens数据集
python main.py
运行结果:
Data Size: n_user = 943, m_item = 1682, cnt = 99057
Device = cuda
Epoch 0: AUC = 0.5002410103152415
Epoch 1: loss = 2.772585, AUC = 0.495324
Epoch 2: loss = 2.772536, AUC = 0.492871
Epoch 3: loss = 2.772536, AUC = 0.494617
...
Epoch 199: loss = 1.054136, AUC = 0.825487
Epoch 200: loss = 1.133315, AUC = 0.828486