Skip to content

Instantly share code, notes, and snippets.

@khuyentran1401

khuyentran1401/data_engineering.py

Last active November 5, 2021 19:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save khuyentran1401/b6aa684c0cc553a0faf8d42b7adef958 to your computer and use it in GitHub Desktop.
Save khuyentran1401/b6aa684c0cc553a0faf8d42b7adef958 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data_engineering.py
# 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