| tags: | |
| - poi | |
| - rqvae | |
| # RQ-VAE (NYC) | |
| Generated at: 2025-10-23T17:51:11.023199+00:00Z | |
| ## Dataset | |
| - Name: NYC | |
| - POIs: 5135 | |
| - Input dim: 1407 | |
| - Feature dims: {'category': 209, 'region': 91, 'temporal': 24, 'collaborative': 1083} | |
| ## Model | |
| - VAE hidden dims: [128, 512, 1024] | |
| - Codebooks: 3 x 32 x 64 | |
| - Commitment weight: 0.5 | |
| - Loss weights: {'reconstruction': 0.0, 'quantization': 1.0, 'utilization': 0.25, 'compactness': 0.25} | |
| ## Training | |
| - Batch size: 128 | |
| - Epochs: 50 | |
| - LR: 5e-05 | |