Skip to content
/ PromptST Public

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

9 stars 1 fork Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

Zhang-Zijian/PromptST

BranchesTags

Folders and files

NameName
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

About

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

Resources

Readme
Activity

Stars

9 stars

Watchers

1 watching

Forks

1 fork
Report repository

Releases

No releases published

Packages

No packages published

Languages