def split_data(data_df, split_pct=0.8):
    """
    Split a DF into two parts based on split_pct
    """
    # Shuffle and split
    split_idx = int(len(data_df) * split_pct)

    data_df  = data_df.sample(frac=1.).reset_index(drop=True)
    train_df = data_df.iloc[:split_idx]
    test_df  = data_df.iloc[split_idx:]

    return train_df, test_df