schema: '2.0'
stages:
  prepare:
    cmd: python3 script_prepare.py data/Car_Prices_Poland_Kaggle.csv
    deps:
    - path: data/Car_Prices_Poland_Kaggle.csv
      md5: 9170e9b525149cb1f571f318cd604913
      size: 9894367
    - path: script_prepare.py
      md5: f1dfe33a503f5acc687c53dee448f71b
      size: 1899
    outs:
    - path: data/Car_Prices_Poland_Kaggle_dev.csv
      md5: cf9355749edc79f588e264de5b2bf1f0
      size: 1648309
    - path: data/Car_Prices_Poland_Kaggle_test.csv
      md5: cf9355749edc79f588e264de5b2bf1f0
      size: 1648309
    - path: data/Car_Prices_Poland_Kaggle_train.csv
      md5: 8818f758e2de344a4b9ad712379b81e1
      size: 6597472
  train:
    cmd: python3 lab05_deepLearning.py 50
    deps:
    - path: data/Car_Prices_Poland_Kaggle_dev.csv
      md5: cf9355749edc79f588e264de5b2bf1f0
      size: 1648309
    - path: data/Car_Prices_Poland_Kaggle_test.csv
      md5: cf9355749edc79f588e264de5b2bf1f0
      size: 1648309
    - path: data/Car_Prices_Poland_Kaggle_train.csv
      md5: 8818f758e2de344a4b9ad712379b81e1
      size: 6597472
    outs:
    - path: CarPrices_pytorch_model.pkl
      md5: cff6a79945bbf839058a4fd1b2dcc98f
      size: 30039
    - path: prediction_results.csv
      md5: 62b9e54cdfebc7f1dfb060e18e9b8738
      size: 585197
  evaluate:
    cmd: python3 lab10_evaluate.py
    deps:
    - path: CarPrices_pytorch_model.pkl
      md5: cff6a79945bbf839058a4fd1b2dcc98f
      size: 30039