Wide and deep report
What ran
- Loaded MovieLens: 全量数据.
- Converted ratings into binary labels:
rating >= 4.0 means positive.
- Used user ID, movie ID, movie genres, and timestamp hour bucket.
- DataLoader workers:
8.
- Device used:
mps.
Metrics
best_valid_logloss: 0.5374best_valid_auc: 0.8049best_valid_accuracy: 0.7268epochs_ran: 16.0000
Examples
Top scored held-out rows:
| userId |
title |
genres |
rating |
predicted_like_probability |
| 6 |
Lord of the Rings: The Fellowship of the Ring, The (2001) |
Adventure |
Fantasy |
3 |
| 6 |
Indiana Jones and the Temple of Doom (1984) |
Action |
Adventure |
Fantasy |
| 14 |
Crackerjack (2002) |
Comedy |
5 |
0.998788 |
| 6 |
Dances with Wolves (1990) |
Adventure |
Drama |
Western |
| 6 |
Election (1999) |
Comedy |
3 |
0.997379 |
| 6 |
Planet of the Apes (1968) |
Action |
Drama |
Sci-Fi |
| 6 |
Star Wars: Episode I - The Phantom Menace (1999) |
Action |
Adventure |
Sci-Fi |
| 14 |
Brewster's Millions (1985) |
Comedy |
5 |
0.99518 |
| 14 |
One Crazy Summer (1986) |
Comedy |
5 |
0.994031 |
| 14 |
Stars and Bars (1988) |
Action |
Comedy |
Romance |
Checkpoint size
| file |
size MB |
best.pt |
67.70 |