n_samples in train test split code example Example: train_test_split example train_test_split example