Entrypoints:
problemframing-eda-featureengineering.ipynb
: EDA, problem definition, and feature engineeringbaseline_models.ipynb
: Training and tuning of baseline models (feature-based models)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