search

WenjieDu / SAITS

The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516
https://doi.org/10.1016/j.eswa.2023.119619
MIT License
309 stars 49 forks source link

Inquiry About Using SAITS Model with 'physionet_2012' Dataset #41

Closed Huozhiyuan closed 3 weeks ago

Huozhiyuan commented 1 month ago

Hi everyone:

While using the 'physionet_2012' dataset, I found that the 'y' values cannot be output as expected. I noticed that the article's 'Downstream classification task' section addresses relevant content. If the 'In-hospital_death' values are not 0 and 1 but some other numbers, and the 'X' section content influences the mortality rate, can the SAITS model be used to predict this? If so, which function of SAITS should I use?

Thank you for your assistance.

Best regards,

WenjieDu commented 1 month ago

Hi there,

Thank you so much for your attention to SAITS! If you find SAITS is helpful to your work, please star⭐️ this repository. Your star is your recognition, which can let others notice SAITS. It matters and is definitely a kind of contribution.

I have received your message and will respond ASAP. Thank you again for your patience! 😃

Best,
Wenjie

Rajesh90123 commented 1 month ago

I am assuming he has used the method as used here for BRITS: https://github.com/WenjieDu/PyPOTS/blob/main/pypots/classification/brits/model.py But yes, I was having the same issue as I could not find the downstream classification task codes in either this github repo or the Awesome Imputation repo. I would be grateful if you could add those @WenjieDu 🙏

github-actions[bot] commented 1 month ago

This issue had no activity for 14 days. It will be closed in 1 week unless there is some new activity. Is this issue already resolved?

WenjieDu commented 1 month ago

@Rajesh90123 Here is the script running a simple RNN on the downstream classification task https://github.com/WenjieDu/SAITS/blob/main/Simple_RNN_on_imputed_data.py

Regarding the Awesome_Imputation repo, we'll add the downstream task scripts recently. Stay tuned, please! You can star and watch it to receive the latest updates ;-) https://github.com/WenjieDu/Awesome_Imputation

github-actions[bot] commented 3 weeks ago

This issue had no activity for 14 days. It will be closed in 1 week unless there is some new activity. Is this issue already resolved?