from flask import Flask, jsonify, render_template, request import tmdbsimple as tmdb from yamr.dataset import EnhancedDataset from yamr.recommender import RecommendationEngine import tmdb_util import util app = Flask(__name__) tmdb.API_KEY = tmdb._get_env_key('TMDB_API_KEY') dataset_path = 'datasets/ml-latest-enhanced' ds = EnhancedDataset(dataset_path) recommender = RecommendationEngine(ds) recommender.load_model('item_sim_model') @app.route('/') def index(): return render_template('index.html') @app.route('/api/movies/<int:movieId>') def movie_detail(movieId): movie = ds.find_movie_by_id(movieId) movie = tmdb_util.add_trailer(movie) return jsonify(movie) @app.route('/api/movies/top_rated')
from yamr.dataset import EnhancedDataset from yamr.recommender import RecommendationEngine, ItemSimilarityRecommender, FactorizationRecommender, PopularityRecommender dataset_path = 'datasets/ml-latest-enhanced' ds = EnhancedDataset(dataset_path) #recommender = ItemSimilarityRecommender() #recommender = FactorizationRecommender() recommender = PopularityRecommender() re = RecommendationEngine(ds, recommender=recommender) re.train_model() re.save_model('popularity_model')