how to do train_test_split with date code example Example: train_test_split example train_test_split example