train-test-split code snippets in python

train test split sklearn template in python