split.py

Used to split the original csv data into train and test parts