openclimatefix / power_perceiver

Machine learning experiments using the Perceiver IO model to forecast the electricity system (starting with solar)
MIT License
7 stars 1 forks source link

Don't reshape_time_as_batch in `GSPQueryGenerator`! #162

Open JackKelly opened 2 years ago

JackKelly commented 2 years ago

Instead, always return (example, time, query_dim), and reshape to (example * time, 1, dim) in SatelliteTransformer.forward, just after going through the gsp_query_generator. But need to ensure we mask future "actual" data!

Then we can simply add in history (if we want it) in the calling code. And then reshape.

And do #144