Unexpected prediction in FFN

I am trying to predict a 5D vector from a 6D vector using a Feed Forward Network.

here is the notebook link: https://colab.research.google.com/drive/1dAIE8-OqLqF0IW_BhxM7L-TBjQOByD-P?usp=sharing

Thanks

I am getting the prediction capped at 1 for some reason