I am trying to implement the neural network model in Keras but I am getting the dimensionality issue. As per the Model architecture, I should get 1 as the output dimension f