CODE EXAMPLE FOR PYTHON

how to split a dataframe into train and test

# Dataframe splitting helper function
def SplitDataframe(df, y_column, test_size=3):
    train_count = int(round(test_size*10/len(df)*100))
    
    train_ds = df[train_count:]
    test_ds = df[:train_count]
    
    train_ds_X = train_ds.drop([y_column], axis=1)
    train_ds_y = train_ds[y_column]
    
    test_ds_X = test_ds.drop([y_column], axis=1)
    test_ds_y = test_ds[y_column]
    
    return (train_ds_X, train_ds_y), (test_ds_X, test_ds_y)