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

Zhang-Zijian / PromptST Public

Notifications You must be signed in to change notification settings
Fork 1
Star 10

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

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
.idea		.idea
__pycache__		__pycache__
data_process		data_process
ft_19.sh		ft_19.sh
ft_4.sh		ft_4.sh
full_19.sh		full_19.sh
full_4.sh		full_4.sh
main.py		main.py
main_fulltrain.py		main_fulltrain.py
pmt_19_st.sh		pmt_19_st.sh
pmt_4_st.sh		pmt_4_st.sh
promptST.py		promptST.py
readme.txt		readme.txt
sin_19.sh		sin_19.sh
sin_4.sh		sin_4.sh
utils.py		utils.py

Repository files navigation

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