Skip to content

SASRec report

What ran

  • Loaded MovieLens: 2,000,000 条评分.
  • Kept high-rating movies as positive user sequences.
  • Used max sequence length 50 and max 20 training examples per user.
  • DataLoader workers: 8.
  • Device used: mps.

Metrics

  • best_valid_loss: 7.7595
  • best_valid_recall@10: 0.0307
  • best_valid_ndcg@10: 0.0143
  • test_loss: 7.9499
  • test_recall@10: 0.0260
  • test_ndcg@10: 0.0121
  • epochs_ran: 11.0000

Examples

history true_next top_recommendations
Emma (1996) -> Living in Oblivion (1995) -> Titanic (1997) Raging Bull (1980) Schindler's List (1993)
Star Wars: Episode IV - A New Hope (1977) -> Schindler's List (1993) -> While You Were Sleeping (1995) -> Ten Commandments, The (1956) -> Superman III (1983) -> Superman IV: The Quest for Peace (1987) -> Matrix Reloaded, The (2003) Shadowlands (1993) X-Men (2000)
Alien (1979) -> Blues Brothers, The (1980) -> The Butterfly Effect (2004) Serenity (2005) Die Hard (1988)
Raiders of the Lost Ark (Indiana Jones and the Raiders of the Lost Ark) (1981) -> Beauty and the Beast (1991) -> Lion King, The (1994) -> Mulan (1998) Birds, The (1963) Lion King, The (1994)
Icarus (2017) -> Jodorowsky's Dune (2013) -> Guard, The (2011) -> Bo Burnham: Inside (2021) -> Laputa: Castle in the Sky (Tenkû no shiro Rapyuta) (1986) -> Ponyo (Gake no ue no Ponyo) (2008) -> News of the World (2020) -> Licorice Pizza (2021) RRR (2022) Get Out (2017)

Checkpoint size

file size MB
best.pt 10.98