comp5331poi/rqvae-tky-div0.1-commit0.25-lr5e-5

RQ-VAE (TKY)

Generated at: 2025-10-23T07:33:57.569936+00:00Z

Dataset

• Name: TKY
• POIs: 7873
• Input dim: 2566
• Feature dims: {'category': 190, 'region': 59, 'temporal': 24, 'collaborative': 2293}

Model

• VAE hidden dims: [128, 512, 1024]
• Codebooks: 3 x 64 x 64
• Commitment weight: 0.25
• Loss weights: {'reconstruction': 1.0, 'quantization': 1.0, 'utilization': 0.1, 'compactness': 0.1}

Training

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