search

microsoft / subseasonal_toolkit

Subseasonal forecasting models
MIT License
43 stars 21 forks source link

Informer baseline runtime error #6

Closed xuchengong closed 1 year ago

xuchengong commented 1 year ago

Dear authors, thank you for your work and sharing! When I run informer.ipynb and informer.py, I encounter a shape incompatibility error in the model's forward, and to solve it I locate at subseasonal_toolkit/models/informer/data/data_loader.py line 329:

df_raw = data_loaders.get_ground_truth("us_" + self.target, sync=False).reset_index()

After I change "us_" + self.target to the variable gt_id, it works. May I ask whether my modification is correct? Please feel free to point out any of my misunderstanding of the dataset or toolkit. Many thanks!

lmackey commented 1 year ago

Thank you for bringing this to our attention. The latest commit to subseasonal_toolkit/models/informer/data/data_loader.py should fix this issue!