Walmart Inc.
SIMILAR ITEM RECOMMENDATION FRAMEWORK USING WIDE-AND-DEEP-ARCHITECTURE

Abstract:

A system including one or more processors and one or more non-transitory computer-readable media storing computing instructions that, when executed on the one or more processors, cause the one or more processors to perform functions comprising: receiving one or more vectors representing one or more types of features for a pair of items; generating, using a similarity item model of a machine learning architecture, a prediction for a similar item, wherein the similarity item model combines a pair of separately trained machine learning models; combining a first output of the gradient boosted model and a second output of the neural network model to generate a similarity score for the pair of items; and transmitting the similar item to a first position on a carousel display of a website that concurrently displays the anchor item on the website. Other embodiments are disclosed.