Skip to content

Recommendation Engine

Jump to bottom
Aditi edited this page Jun 11, 2019 · 6 revisions

Description

  • The Recommendation Engine for this project was trained using SVD algorithm in Scikit-Surprise.
  • RecommendationEngine stores trained model in form of pickle file, which can be stored in database.The pickle file can be used to make predictions for a new user.

Setup

**1. ImportRecommendationEngine and create an instance of the class. Use the instance to train SVD model and generate top n recommendations for new and existing users.

model = RecommendationEngine()
model.train_svd()
model.get_top_n(10)

Clone this wiki locally