Dataset | Parameter | Value |
---|---|---|
TripAdvisor | User and item embedding vector size | 256 |
Hidden layers | [512 → 256 → 128 → 64 → 32 → 16 → 8] | |
Output layer | 1 | |
Batch size | 512 | |
Movies dataset | User and item embedding vector size | 128 |
Hidden layers | [256 → 128 → 64 → 32 → 16 → 8] | |
Output layer | 1 | |
Batch size | 64 |