search

ant-research / Pyraformer

Apache License 2.0
237 stars 37 forks source link

How to get the forecast result picture #11

Open Common-Player opened 2 years ago

Common-Player commented 2 years ago

Hello, Author Thanks for the your great idea is in the field of time series prediction.When I run your code, I have some trouble. I really hope to get your help when I run the single_step_main.py, I find I get a best_model.pth in folder named SingleStep,but i don't know how to view the forecast results of this model.The best thing is that I can intuitively understand the prediction results through the pictures. What should I do?

Zhazhan commented 2 years ago

We are sorry that we did not provide visualization code in the current release. We will add visualization functions in the next version. You can also visualize the results using matplotlib in single_step_main.py.

Common-Player commented 2 years ago

Thank you very much for your reply! I have some more questions.In the main function of this file, which variable is used to save the prediction results?If I can know this, I will be able to do visualization through Matplotlib.Thank you again for your reply!

Zhazhan commented 2 years ago

In line 143 of single_step_main.py, "mu_pre" stores the predicted results of the model. You can visualize it together with "label" to see the model prediction performance. I hope this can help you.

Common-Player commented 2 years ago

In line 143 of single_step_main.py, "mu_pre" stores the predicted results of the model. You can visualize it together with "label" to see the model prediction performance. I hope this can help you.

OK!I know what I should do. Thank you again!

blueaaa7 commented 1 year ago

Hello, could you tell me how to output predicted images?

In line 143 of single_step_main.py, "mu_pre" stores the predicted results of the model. You can visualize it together with "label" to see the model prediction performance. I hope this can help you.

OK!I know what I should do. Thank you again!