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
Zhang-Zijian/PromptST
This repo is for the implementation of CIKM 2023 paper: "Prompt-Enhanced Spatio-Temporal Multi-Attribute Prediction"
Python