В работе были использованы:
- Данные из соревнования от H&M https://www.kaggle.com/competitions/h-and-m-personalized-fashion-recommendations/overview
- модель Pinsage https://arxiv.org/pdf/1806.01973.pdf
- Реализация модели от разработчтков dgl https://github.com/dmlc/dgl/tree/master/examples/pytorch/pinsage.
Результаты:
Модель обучалась 41 эпоху, итоговое значение функции потерь: 0.3630964728664607
Пример:
Исследуемый товар
Рекоммендации / топ 3 схожих товаров