NCF report
What ran
- Loaded MovieLens: 全量数据.
- Converted ratings into binary labels:
rating >= 4.0 means positive.
- Used user ID and movie ID.
- DataLoader workers:
8.
- Device used:
mps.
Metrics
best_valid_logloss: 0.5308best_valid_auc: 0.8091best_valid_accuracy: 0.7302epochs_ran: 16.0000
Examples
Top scored held-out rows:
| userId |
title |
genres |
rating |
predicted_like_probability |
| 14 |
Crackerjack (2002) |
Comedy |
5 |
0.991771 |
| 50 |
Cool Hand Luke (1967) |
Drama |
4.5 |
0.98562 |
| 40 |
Seven Samurai (Shichinin no samurai) (1954) |
Action |
Adventure |
Drama |
| 14 |
One Crazy Summer (1986) |
Comedy |
5 |
0.981065 |
| 6 |
Lord of the Rings: The Fellowship of the Ring, The (2001) |
Adventure |
Fantasy |
3 |
| 50 |
Clockwork Orange, A (1971) |
Crime |
Drama |
Sci-Fi |
| 50 |
Diving Bell and the Butterfly, The (Scaphandre et le papillon, Le) (2007) |
Drama |
4.5 |
0.977615 |
| 50 |
Trainspotting (1996) |
Comedy |
Crime |
Drama |
| 14 |
Perfectly Normal (1990) |
Comedy |
5 |
0.976074 |
| 50 |
Sleuth (1972) |
Comedy |
Mystery |
Thriller |
Checkpoint size
| file |
size MB |
best.pt |
66.59 |