Skip to content

Instantly share code, notes, and snippets.

What would you like to do?
from prefect import task, Flow
with Flow("data-engineer") as flow:
# Define parameters
target_col = 'species'
test_data_ratio = 0.2
# Define tasks
data = load_data(path="data/raw/iris.csv")
classes = get_classes(data=data, target_col=target_col)
categorical_columns = encode_categorical_columns(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