Prediction of the investor's class by transactions and the initial state of the portfolio
| Название | Содержание |
|---|---|
| train_models.py | главный скрипт |
Данные взяты из чемпионата инвесторов, где каждый из инвесторов в итоге занял какое-то место. Все инвесторы согласно этим местам были поделены на 5 классов. Нам нужно было научиться предсказывать этот класс по сделкам и начальному состоянию портфеля.
В решении была использованна модель LSTM для предсказания класс инвестора по его сделкам, а финальная классификация производится с помощью модели Catboost, с агрегированными фичами, предсказаниями модели LSTM и исходными данными
Telegram: @suhowandrew, @nizovtsevanastya VK: vk.com/suhowandrew, vk.com/tusik_mrr,