Work around the StandardScaler broadcasting error in modern (1.0+) versions of sklearn

QData / spacetimeformer

Multivariate Time Series Forecasting with efficient Transformers. Code for the paper "Long-Range Transformers for Dynamic Spatiotemporal Forecasting."

https://arxiv.org/abs/2109.12218

MIT License

808 stars 191 forks source link

Work around the StandardScaler broadcasting error in modern (1.0+) versions of sklearn #29

Closed jakegrigsby closed 2 years ago

jakegrigsby commented 2 years ago

Fixes common issue (#8, #16) caused by sklearn shape assert statements introduced between the version I used to develop this and the recent versions I should've used to begin with...