Without using an OutputProjectionWrapper
0 MSE: 13.907031
100 MSE: 0.5056713
200 MSE: 0.20992868
300 MSE: 0.10282391
400 MSE: 0.06876178
500 MSE: 0.061076507
600 MSE: 0.054797072
700 MSE: 0.04966834
800 MSE: 0.050121456
900 MSE: 0.047807403
1000 MSE: 0.04775478
1100 MSE: 0.050040428
1200 MSE: 0.041216638
1300 MSE: 0.05031095
1400 MSE: 0.04354385
y_prediction:
[[[-3.4392304 ]
[-2.4754486 ]
[-1.1431999 ]
[ 0.67572594]
[ 2.2022252 ]
[ 3.0979629 ]
[ 3.4667068 ]
[ 3.317198 ]
[ 2.856221 ]
[ 2.2434225 ]
[ 1.7069325 ]
[ 1.5460964 ]
[ 1.8807409 ]
[ 2.7058399 ]
[ 3.893656 ]
[ 5.112901 ]
[ 6.1083117 ]
[ 6.658971 ]
[ 6.645973 ]
[ 6.051144 ]]]
