splitting dataset into train, valid and test sets

splitting-dataset.py

from sklearn.model_selection import train_test_split

data = np.copy(tib)
train_set, test_set = train_test_split(data, test_size=0.3, random_state=42)
train_set, valid_set = train_test_split(train_set, test_size=0.2, random_state=42)

# printing the shapes of the datasets
train_set.shape, valid_set.shape, test_set.shape
'''
Out:
((3360, 4), (840, 4), (1800, 4))
'''