Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 626 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 626 Bytes

Entrypoints:

  1. problemframing-eda-featureengineering.ipynb: EDA, problem definition, and feature engineering
  2. baseline_models.ipynb: Training and tuning of baseline models (feature-based models)
  3. rnn_models.ipynb: Training of sequence-based models

Modules: rnn_dataset.py, rnn_model.py, rnn_trainer.py, utils.py

If you want to re-create the results, please create a new environment with requirements.txt and put the data files into the ./data/instacart-market-basket-analysis/ folder. The Instacart dataset can be found here: https://www.kaggle.com/competitions/instacart-market-basket-analysis/data