Fix plotting

[Jaeyoung-Lim]

Problem Description Plotting of the predictions were broken as reported in https://github.com/ethz-asl/data-driven-dynamics/issues/180

This was due to the fact that the axis order of the prediction / measurement has been flipped after https://github.com/ethz-asl/data-driven-dynamics/pull/176

This PR fixes the plotting by fixing the order of the matrices

Testing The results can be reproduced with the following command.

make estimate-model plot=True

Before PR

After PR

Additional Context

• This is a follow up of https://github.com/ethz-asl/data-driven-dynamics/pull/181
• Fixes https://github.com/ethz-asl/data-driven-dynamics/issues/180