neurips2026citympc/citympc-txvit

CityMPC — TX_ViT pre-trained checkpoints (5 cities)

Stripped Lightning checkpoints for the CityMPC TX_ViT model from the NeurIPS 2026 paper. One checkpoint per city, selected by the seed-sweep stability filter (Kendall weighting σ_presence < 1e-3 + good-verdict on test-set outlier rule), then ranked by lowest Path Loss MAE / best Presence F1.

Per-city metrics (test set)

How to use

These checkpoints are stripped — inference only, not training-resumable. After cloning the CityMPC repo:

# download a single city
hf download neurips2026citympc/citympc-txvit \
  outputs/tx_austin/city_10_austin_3p5_s/final/ckpts/last.ckpt \
  outputs/tx_austin/city_10_austin_3p5_s/final/config.yaml \
  --local-dir .

# evaluate
python scripts/analyze.py \
  --ckpt outputs/tx_austin/city_10_austin_3p5_s/final/ckpts/last.ckpt

Or open notebooks/02_analyze_trained_model.ipynb and paste the downloaded run path into TXVIT_RUN_DIR.

File layout

outputs/
  tx_austin/city_10_austin_3p5_s/final/{ckpts/last.ckpt, config.yaml}
  tx_dallas/city_8_dallas_3p5_s/final/{ckpts/last.ckpt, config.yaml}
  tx_denver/city_18_denver_3p5_s/final/{ckpts/last.ckpt, config.yaml}
  tx_fortworth/city_12_fortworth_3p5_s/final/{ckpts/last.ckpt, config.yaml}
  tx_newyork/city_0_newyork_3p5_s/final/{ckpts/last.ckpt, config.yaml}

Integrity (SHA-256)

Notes

• —Architecture: TX_ViT conditional VAE (CityMPC). See the paper for details.
• —Trained on the CityMPC dataset.
• —Lightning state stripped (optimizer / scheduler / callback) — checkpoints are not training-resumable.
• —σ-gate: every shipped run had weighting/sigma/presence < 1e-3 at its last logged step.