Customer Behaviour Prediction Github, model_selection import train_test_split from sklearn.