Using an OuputProjectionWrapper
0 MSE: 11.9672575
100 MSE: 0.52584445
200 MSE: 0.14959829
300 MSE: 0.07381975
400 MSE: 0.06171744
500 MSE: 0.05968452
600 MSE: 0.055536177
700 MSE: 0.047983035
800 MSE: 0.050073363
900 MSE: 0.04728166
1000 MSE: 0.047391903
1100 MSE: 0.048076287
1200 MSE: 0.040791243
1300 MSE: 0.0479767
1400 MSE: 0.04211445
y_prediction
[[[-3.406515 ]
[-2.4501145]
[-1.1305888]
[ 0.7817146]
[ 2.2008858]
[ 3.138787 ]
[ 3.399897 ]
[ 3.3627138]
[ 2.8842826]
[ 2.2652197]
[ 1.650614 ]
[ 1.5299088]
[ 1.8965788]
[ 2.731001 ]
[ 3.9058359]
[ 5.14548 ]
[ 6.144004 ]
[ 6.6781516]
[ 6.6523943]
[ 6.0744486]]]
