qq1349420243's picture
Upload README.md with huggingface_hub
c8f3297 verified
metadata
tags:
  - poi
  - rqvae

RQ-VAE (TKY)

Generated at: 2025-10-23T16:47:23.145181+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.75, 'compactness': 0.75}

Training

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