How to adapt to a different dataset?

[chaitjo]

Thanks for releasing the code! I was wondering if the code is super specific to the Amazon product search data or if it is possible to use it for custom datasets as well?