comp5331poi/Nrqvae-NYC_Exploration-div0.25-commit0.25-lr1e-3

RQ-VAE (NYC)

Generated at: 2025-11-06T19:08:33.136319+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, 256, 512]
• Codebooks: 3 x 32 x 64
• Commitment weight: 0.25
• Loss weights: {'reconstruction': 1.0, 'quantization': 1.0, 'utilization': 0.25, 'compactness': 0.25}

Training

• Batch size: 128
• Epochs: 3000
• LR: 0.001