Zhang-Zijian / PromptST

This repo is for the implementation of CIKM 2023 paper: "Prompt-Enhanced Spatio-Temporal Multi-Attribute Prediction"

Pretrain/Full Train:
  -Complaint: bash full_19.sh
  -NYC Taxi: bash full_4.sh

Finetune:
  -Complaint: bash ft_19.sh
  -NYC Taxi: bash ft_4.sh

Prompt Tune:
  -Complaint: bash pmt_st_19.sh
  -NYC Taxi: bash pmt_st_4.sh
  
Single Train:
  -Complaint: bash sin_19.sh
  -NYC Taxi: bash sin_4.sh