0x2620/dd-re
2
0
Fork 0
Code Issues 4 Pull requests Releases Wiki Activity

added top scoring playlists for user and global keywords to dd-re debug view

Browse source
This commit is contained in:
pythagoraswitch 2018-11-29 21:05:59 +01:00
parent 319e4d384e
commit e3c61853c3
1 changed files with 46 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
recommendation_engine.py
View file

@ -110,14 +110,17 @@ class Engine:
# If the most recent event is "login," initialize grid videos.
if user.get('events', [{}])[0].get("event")=="login":
rec = self.get_recommendations(playlists, user)
return {
'user': {
'keywords': user.get('keywords', {})
},
'videos': self.get_recommendations(playlists, user),
'videos': rec["videos"],
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
"top_user_keywords": top_user_keywords,
"top_user_characters": top_user_characters,
"top_user_playlists": rec["top_user_playlists"],
"top_global_playlists": rec["top_global_playlists"]
}
}
@ -144,14 +147,17 @@ class Engine:
# if there were no grid events for all, initialize all grids.
if len(prev_grid_list) < video_num:
rec = self.get_recommendations(playlists, user)
return {
'user': {
'keywords': user.get('keywords', {})
},
'videos': self.get_recommendations(playlists, user),
'videos': rec["videos"],
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
"top_user_keywords": top_user_keywords,
"top_user_characters": top_user_characters,
"top_user_playlists": rec["top_user_playlists"],
"top_global_playlists": rec["top_global_playlists"]
}
}
@ -215,8 +221,8 @@ class Engine:
vids_exclude = [e.get("playlist") for e in prev_grid_list]
while None in vids_exclude:
vids_exclude.remove(None)
video = self.get_recommendations(playlists, user, vids_exclude)
rec_list += [(i, video[i]) for i in next_playlist_index]
rec = self.get_recommendations(playlists, user, vids_exclude)
rec_list += [(i, rec['videos'][i]) for i in next_playlist_index]
rec_list = sorted(rec_list, key=lambda k:k[0])
@ -227,8 +233,10 @@ class Engine:
},
'videos': videos_,
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
"top_user_keywords": top_user_keywords,
"top_user_characters": top_user_characters,
"top_user_playlists": rec["top_user_playlists"],
"top_global_playlists": rec["top_global_playlists"]
}
}
@ -269,6 +277,13 @@ class Engine:
playlists,
key=lambda playlist: -score[playlist['name']]
)
# Record the following for debug view input
top_user_playlists = [{
'name': playlist['name'],
'tags': playlist['tags']
'score': score[playlist['name']]
} for playlist in playlists[:channels['userKeywords']]]
videos = playlists[:channels['userKeywords']]
playlists = playlists[channels['userKeywords']:]
# For each playlist, compute global keyword score
@ -282,15 +297,26 @@ class Engine:
playlists,
key=lambda playlist: -score[playlist['name']]
)
# Record the following for debug view input
top_global_playlists = [{
'name': playlist['name'],
'tags': playlist['tags']
'score': score[playlist['name']]
} for playlist in playlists[:channels['globalKeywords']]]
videos += playlists[:16 - channels['userKeywords']]
# Shuffle playlists (randomize layout) and shift clips (randomize start)
random.shuffle(videos)
return [{
return {
'videos': [{
'clips': video['clips'],
'position': random.choice([i for i in range(len(video["clips"])) if video["clips"][i].get("pass",False)!=True]),
'position': random.choice([i for i in range(len(video["clips"])) if not video["clips"][i].get("pass")]),
'name': video['name'],
'tags': video['tags'],
} for video in videos]
} for video in videos],
"top_user_playlists":top_user_playlists,
"top_global_playlists": top_global_playlists
}
def update_user_playlists(self, user, watch_cutoff = 0.9):
@ -349,7 +375,7 @@ class Engine:
def get_next(self, user, position):
# Update self_playlists first to reflect changes
# Update self_playlists to reflect user log history
playlists = self.update_user_playlists(user)
grid_events = {}
@ -362,8 +388,8 @@ class Engine:
break
prev_grid_list = sorted([v for v in grid_events.values()], key=lambda k:k['index'])
vids_exclude = [e.get("playlist") for e in prev_grid_list]
video = self.get_recommendations(playlists, user, vids_exclude)[position]
return video
rec = self.get_recommendations(playlists, user, vids_exclude)
return rec["videos"][position]
def update_state(self, data):
for key in data: