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

added top scoring keywords to debug view output

Browse source
This commit is contained in:
pythagoraswitch 2018-11-20 02:22:09 +01:00
parent 001e377003
commit 2405b8b7c7
1 changed files with 21 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
recommendation_engine.py
View file

@ -97,13 +97,23 @@ class Engine:
def get_videos(self, user):
# Update self_playlists first to reflect changes
playlists = self.update_user_playlists(user)
# Get the user keyword scores for debug view
user_keywords = copy.deepcopy(user.get('keywords', {}))
theme_tags = {k.lower():v for k,v in user_keywords.items() if not k.isupper()}
character_tags = {k:v for k,v in user_keywords.items() if k.isupper()}
top_user_keywords = sorted([(k,v) for (k,v) in theme_tags.items()], key=lambda kv: kv[1])[-5:]
top_user_characters = sorted([(k,v) for (k,v) in character_tags.items()], key=lambda kv: kv[1])[-5:]
if user.get('events', [{}])[0].get("event")=="login":
return {
'user': {
'keywords': user.get('keywords', {})
},
'videos': self.get_recommendations(playlists, user)
'videos': self.get_recommendations(playlists, user),
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
}
}
channels = {k: v.get('value', 0) for k, v in self.state['channels'].items()}
@ -133,7 +143,11 @@ class Engine:
'user': {
'keywords': user.get('keywords', {})
},
'videos': self.get_recommendations(playlists, user)
'videos': self.get_recommendations(playlists, user),
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
}
}
else:
@ -211,7 +225,11 @@ class Engine:
'user': {
'keywords': user.get('keywords', {})
},
'videos': videos_
'videos': videos_,
"_debug": {
"top_user_keywords":top_user_keywords,
"top_user_characters":top_user_characters
}
}