diff --git a/recommendation_engine.py b/recommendation_engine.py index e7534a3..59ea1d3 100644 --- a/recommendation_engine.py +++ b/recommendation_engine.py @@ -78,8 +78,8 @@ class Engine: return clips def get_videos(self, user): - channels = {'keywords': 7, 'screenings': 7, 'random': 2} - sliders = {'dau': -1, 'physics': 0, 'sex': 1} + channels = self.state['channels'] + sliders = self.state['keywords'] # For each playlist, compute keyword score score = {} for playlist in self.playlists: