Trading armatures

Модели: data/

Реализована модель для прогнозирования временнного ряда и алгоритм отбора оптимальных объемов закупки товара.

Установка и запуск

Clone the repo and change to the project root directory:

git clone https://github.com/AGoldian/rl_armatures
cd rl_armatures

Install requirements:

pip install -r req.txt

And run:

python main.py

Используемое решение

Были использованы несколько подходов к решению задачи:

• EDA: основная проблема и ограничения, данные были неполные и множество были невалидными
• Выбор архитектуры: пробовались варианты RL (elegant RL), RNN на основе LSTM и градиентный бустинг Catboost. Остановили выбор на сверточных сетях с слоями LSTM, так как в следствии существует выход на RL.
• Алгоритм покупки: реализация оптимального trading-алгоритма для покупки товара в зависимости от временного ряда