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

only count events with product

Browse source
This commit is contained in:
j 2018-02-04 16:05:55 +01:00
parent 935ea07559
commit 49e045daf4
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
recommendation_engine.py
View file

@ -65,7 +65,8 @@ class Engine:
# Count tags for the user # Count tags for the user
count = defaultdict(lambda: 0) count = defaultdict(lambda: 0)
for event in user.get('events', []): for event in user.get('events', []):
count[event['data']['product']] += 1 if event.get('data', {}).get('product'):
count[event['data']['product']] += 1
# For each tag in playlist, increment score by count # For each tag in playlist, increment score by count
for playlist in playlists: for playlist in playlists:
score[playlist['name']] = random.random() score[playlist['name']] = random.random()