Attention-based neural re-ranking approach for next city in trip recommendations

03/23/2021
by   Aleksandr Petrov, et al.
0

This paper describes an approach to solving the next destination city recommendation problem for a travel reservation system. We propose a two stages approach: a heuristic approach for candidates selection and an attention neural network model for candidates re-ranking. Our method was inspired by listwise learning-to-rank methods and recent developments in natural language processing and the transformer architecture in particular. We used this approach to solve the Booking.com recommendations challenge Our team achieved 5th place on the challenge using this method, with 0.555 accuracy@4 value on the closed part of the dataset.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset