I am specifically doing this way since i need X_train, X_test, y_train, y_test else i could have easily fit with X,y. But i am not sure whether below mentioned code is the right