Вы можете использовать нижеприведенный код для создания тестовых и обучающих образцов:
from sklearn.model_selection import train_test_split
trainingSet, testSet = train_test_split(df, test_size=0.2)
Размер теста может варьироваться в зависимости от процента данных, которые вы хотите разместить в тестовом наборе данных.