here it is for the testing set

its learning cause at the beginning of the training, i get different shapes, with some tweaks with the hyper parameters, I manage to get what i sent before.
maybe this could give you a better idea, i tried different parameters, and i got the following