Skip to content

Latest commit

 

History

History
31 lines (20 loc) · 1.02 KB

README.md

File metadata and controls

31 lines (20 loc) · 1.02 KB

Predict Future Sales

My attempt to solve Kaggle's Predict Future Sales competition.

Contains:

  • RNN approach (pytorch)
  • Data processing script in Spark (scala)
  • Creation of embeddings for item/item category/shop descriptions
  • Feature importance check using eli5
  • And more!

Report

For the solution report, check here.

Automatic hyperparameter tuning

For RNN-based model, automatic hyperparemeter tuning is available with Guild AI AutoML tool.

Check this document for details.

External resources

The default location for these is the data folder.

Dependencies

All you need is included in environment.yml. Conda is the package manager for this project.