Congratulate on your article! Could you provide the code for split data into X_train, Y_train, X_test, Y_test ?

Thanks