def load_split_csv_data()

in Project-AutoML/automl/data.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def load_split_csv_data(data_path, label_column, test_size=0.25, random_state=1):

    data = pd.read_csv(data_path)
    train, test = train_test_split(data, test_size=test_size, random_state=random_state)
    train_x = train.drop([label_column], axis=1)
    test_x = test.drop([label_column], axis=1)
    train_y = train[[label_column]]
    test_y = test[[label_column]]
    return train_x, train_y, test_x, test_y