It’s a little difficult to see what the exact problem is without knowing what the format of x_train, and y_train, etc. are when creating the dataset. It could just be that Y is a wrapped version of the data. You can check this via len(X1), len(X2), len(Y); if one is length 1 then the data needs to be unwrapped.