stages:
  prepare:
    cmd: python3 process_dataset.py
    deps:
    - lego_sets.csv
  train:
    cmd: python3 simple_regression.py 100
    deps:
    - lego_sets_clean_train.csv
    - lego_sets_clean_test.csv