Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# Define parameters
target_col = 'species'
test_data_ratio = 0.2
# Run functions
data = load_data(path="data/raw/iris.csv")
categorical_columns = encode_categorical_columns(data=data, target_col=target_col)
classes = get_classes(data=data, target_col=target_col)
train_test_dict = split_data(data=categorical_columns,
test_data_ratio=test_data_ratio,
classes=classes)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment