跳转至

NGCF 实验报告

本次运行做了什么

  • 读取 MovieLens:全量数据。
  • 用 rating >= 4.0 构建用户-电影二部图。
  • 每个用户最多保留 50 条最近正反馈边;传 0 可使用全部正反馈。
  • 图训练设备:cpu

指标

  • best_valid_loss1.5886
  • best_valid_recall@100.0500
  • best_valid_ndcg@100.0251
  • test_loss1.6412
  • test_recall@100.0420
  • test_ndcg@100.0233
  • epochs_ran6.0000

推荐样例

test_positive top_recommendations
Ever After: A Cinderella Story (1998) Pulp Fiction (1994)
Hunchback of Notre Dame, The (1996) Pulp Fiction (1994)
Sister Act 2: Back in the Habit (1993) Pulp Fiction (1994)
Stir of Echoes (1999) Shawshank Redemption, The (1994)
Lion King, The (1994) Shawshank Redemption, The (1994)

Checkpoint 大小

文件 大小 MB
best.pt 58.11