'NoneType' object is not subscriptable when running 'predict.ipynb'

thuml / Autoformer

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008

MIT License

1.99k stars 429 forks source link

'NoneType' object is not subscriptable when running 'predict.ipynb' #78

Closed erving666 closed 2 years ago

erving666 commented 2 years ago

TypeError Traceback (most recent call last)

in 4 plt.figure() 5 #预测OT ----> 6 plt.plot(prediction[-1,:,-1])#由于prediction.shape是[1,24,7]那么batch只有1 索引只能是0或-1 都是代表batch这一维本身 7 print(prediction.shape) 8 plt.show() TypeError: 'NoneType' object is not subscriptable

wuhaixu2016 commented 2 years ago

Hi, you can set the argument do_predict as True.