Thank you for your reply. This is supposed to be binary classification model. I am trying to learn how to implement an embedding layer. As for how to fit the target shape, this is exactly what I want to know!! Why the prediction become 3 dimensional ?