Skip to content

Restaurant Recommendation Systems based on the Yelp dataset using Wide and Deep Recommendation System, Infersent (Annoy indexed), ALS, Factor Machines and Graph based models.

License

Notifications You must be signed in to change notification settings

AmoghM/Yelp-Restaurants-RecSys

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

65 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Steps to follow for recommendation on Yelp dataset

a) Pre-run step:

  • Clone the repository
  • Install packages and dependencies: pip install -r requirements.txt
  • Download Yelp dataset from here: https://www.yelp.com/dataset/challenge
  • Place the extracted folder into the data/ of the repository
  • Run Data_Pre_Processing.ipynb from src folder. This will create the relevant datasets necessary to run the models we have defined

b) For Bias Baseline and ALS model:

  • Run src/ALS_Baseline.ipynb

c) For Factorization Machine model:

  • Run src/CMF_FM.ipynb

d) For Wide and Deep model:

  • Run src/Wide and Deep.ipynb

e) For Content-based recommendation:

  • Download glove, infersent model mentioned here: https://github.com/facebookresearch/InferSent
  • Move the infersent2.pkl to src/Content-Recommendation
  • Run python src/json_to_csv.py to convert json to csv consisting of Las Vegas restaurant dataset for 2018.
  • Run python src/Content-Recommendation/get_review_embedding.py to generate weighted review2vec and export it to a file.
  • Run python src/Content-Recommendation/content_recommendation.py to create annoy index from review embeddings and provide top 10 recommendations for the input string.

Models Used:

Team

L-R: Benjamin, Siddhant, Swarna and Amogh team

About

Restaurant Recommendation Systems based on the Yelp dataset using Wide and Deep Recommendation System, Infersent (Annoy indexed), ALS, Factor Machines and Graph based models.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published