search

NicolasHug / Surprise

A Python scikit for building and analyzing recommender systems
http://surpriselib.com
BSD 3-Clause "New" or "Revised" License
6.34k stars 1.01k forks source link

Binary recommendation #412

Open VishnuBhaarath opened 2 years ago

VishnuBhaarath commented 2 years ago

Hi, I am working on a binary recommendation system I want to predict whether an user buys a product or not, 0 fornot buying a product and 1 for buying it. I have gone through the (surprise) documentation and have set up the reader class with a rating_scale of 0 to 1 Reader(rating_scale=(0,1)), I have trained it using SVD and for the corresponding user id and product id I get a value between 0 and 1, so is it advisable to set some threshold like 0.5 or something(values less than it are consider as 0 and above or considered as 1)?Or is their any other method to approach this problem. Thanks in advance.

NicolasHug commented 2 years ago

Hi @VishnuBhaarath , surprise isn't really suited for binary recommendation, you might want to check out https://github.com/benfred/implicit instead