Solved by ptrblck in post #2 
              
              
                Could you check the size of y_train? 
While X_train seems to contain 487680 samples, it’s not clear if y_train might be smaller.