Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add pypots.forecasting.Transformer #597

Merged
merged 6 commits into from
Mar 5, 2025
+674 −8
Merged

Add pypots.forecasting.Transformer #597

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
6f513ad
fix: X_pred_missing_mask fetched from X_pred by mistake;
WenjieDu Mar 4, 2025
dff8137
feat: make MSE as forecasting models' default training loss and eval …
WenjieDu Mar 4, 2025
fff9585
fix: wrong target in validation stage;
WenjieDu Mar 4, 2025
69100b2
feat: enable Transformer to work on the forecasting task;
WenjieDu Mar 4, 2025
b31e16f
docs: update docs for forecasting Transformer;
WenjieDu Mar 4, 2025
8fe0bd6
test: add testing cases for forecasting Transformer;
WenjieDu Mar 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
fix: X_pred_missing_mask fetched from X_pred by mistake;
@WenjieDu
WenjieDu committed Mar 4, 2025
commit 6f513ad03bd0e49bc5a348d5dcba776f054de23e
4 changes: 2 additions & 2 deletions pypots/data/dataset/base.py
View file
Original file line number Diff line number Diff line change
@@ -342,8 +342,8 @@ def _fetch_data_from_array(self, idx: int) -> Iterable:

if self.return_X_pred:
X_pred = self.X_pred[idx]
pred_missing_mask = self.X_pred[idx]
sample.extend([X_pred, pred_missing_mask])
X_pred_missing_mask = self.X_pred_missing_mask[idx]
sample.extend([X_pred, X_pred_missing_mask])

if self.return_y:
sample.append(self.y[idx].to(torch.long))