comp5331poi/rqvae-nyc-div0.1-commit0.5-lr5e-5

RQ-VAE (NYC)

Generated at: 2025-10-23T07:32:23.049703+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': 1.0, 'quantization': 1.0, 'utilization': 0.1, 'compactness': 0.1}

Training

• Batch size: 128
• Epochs: 50
• LR: 5e-05