v0.2.1
Here are updates,
- for missing values after LOCF imputation (that are missing since the first step hence LOCF doesn't work), we added more options to handle them. Please refer to the argument
first_step_imputation
in LOCF docs. The default option is "zero" in previous versions, but we've changed it to "backward" which is more reasonable; - enabled SAITS to return latent attention weights from blocks in predict() for advanced analysis e.g. in #178;
- renamed model saving and loading functions save_model() and load_model() into save() and load();
What's Changed
- Check if X_intact contains missing data for imputation models, check and list mismatched hyperparameters in the tuning mode by @WenjieDu in #234
- Make SAITS return attention weights in predict() by @WenjieDu in #239
- Adding other options for the first step imputation in LOCF by @WenjieDu in #240
- Fixing the problem about staling issues by @WenjieDu in #244
- Testing with Python 3.11 and support it by @WenjieDu in #246
- Rename save_model() and load_model() into save() and load() by @WenjieDu in #247
- Refactoring save_model() and load_model(), and updating docs by @WenjieDu in #249
Full Changelog: v0.2...v0.2.1