I think from the output that I get it is of shape [250, 1] and not [250, 15]. It should be [250, 15]. I’m not understanding where I have to make the change within the training loop or the evaluation function.