Exactly follow the procedure and run the code. But, there is an issue when train the model ValueError: Error when checking target: expected dense_1 to have 2 dimensions, but got array with shape (40261, 2, 2) How to fix this one!