mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-24 07:53:43 +03:00
185 lines
6.6 KiB
Python
185 lines
6.6 KiB
Python
from annoy import AnnoyIndex
|
|
from .mapping.mapping import *
|
|
from .models.models import *
|
|
from passfinder.events.models import Event
|
|
from passfinder.recomendations.models import UserPreferences
|
|
from random import choice
|
|
from collections import Counter
|
|
|
|
|
|
def get_nearest_(instance_model, model_type, mapping, nearest_n, ml_model):
|
|
how_many = len(Event.objects.filter(type=model_type))
|
|
|
|
index = mapping.index(instance_model.oid)
|
|
nearest = ml_model.get_nns_by_item(index, len(mapping))
|
|
|
|
res = []
|
|
for i in range(how_many):
|
|
try:
|
|
res.append(Event.objects.get(oid=mapping[nearest[i]]))
|
|
except Event.DoesNotExist: ...
|
|
if len(res) == nearest_n: break
|
|
return res
|
|
|
|
|
|
def nearest_attraction(attraction, nearest_n):
|
|
return get_nearest_(attraction, 'attraction', attraction_mapping, nearest_n, attracion_model)
|
|
|
|
|
|
def nearest_movie(movie, nearest_n):
|
|
return get_nearest_(movie, 'movie', cinema_mapping, nearest_n, cinema_model)
|
|
|
|
|
|
def nearest_plays(play, nearest_n):
|
|
return get_nearest_(play, 'plays', plays_mapping, nearest_n, plays_model)
|
|
|
|
|
|
def nearest_excursion(excursion, nearest_n):
|
|
return get_nearest_(excursion, 'excursion', excursion_mapping, nearest_n, excursion_model)
|
|
|
|
|
|
def nearest_concert(concert, nearest_n):
|
|
return get_nearest_(concert, 'concert', concert_mapping, nearest_n, concert_model)
|
|
|
|
|
|
def get_nearest_event(event, nearest_n):
|
|
if event.type == 'plays':
|
|
return nearest_plays(event, nearest_n)
|
|
if event.type == 'concert':
|
|
return nearest_concert(event, nearest_n)
|
|
if event.type == 'movie':
|
|
return nearest_movie(event, nearest_n)
|
|
|
|
|
|
def update_preferences_state(user, event, direction):
|
|
pref = UserPreferences.objects.get(user=user)
|
|
|
|
if direction == 'left':
|
|
if event.type == 'plays':
|
|
pref.unpreffered_plays.add(event)
|
|
if event.type == 'movie':
|
|
pref.unpreffered_movies.add(event)
|
|
if event.type == 'concert':
|
|
pref.unpreferred_concerts.add(event)
|
|
else:
|
|
if event.type == 'plays':
|
|
pref.preffered_plays.add(event)
|
|
if event.type == 'movie':
|
|
pref.preffered_movies.add(event)
|
|
if event.type == 'concert':
|
|
pref.preferred_concerts.add(event)
|
|
pref.save()
|
|
|
|
|
|
|
|
def get_next_tinder(user, prev_event, prev_direction):
|
|
pref = UserPreferences.objects.get(user=user)
|
|
print(prev_event.type, len(pref.preferred_concerts.all()))
|
|
if prev_direction == 'left':
|
|
if prev_event.type == 'plays' and len(pref.unpreffered_plays.all()) <= 2:
|
|
candidates = nearest_plays(prev_event, 100)
|
|
# print(candidates, type(candidates), len(Event.objects.filter(type='plays')))
|
|
return candidates[-1]
|
|
if prev_event.type == 'movie' and len(pref.unpreffered_movies.all()) <= 2:
|
|
candidates = nearest_movie(prev_event, 100)
|
|
return candidates[-1]
|
|
if prev_event.type == 'concert' and len(pref.unpreferred_concerts.all()) <= 2:
|
|
candidates = nearest_concert(prev_event, 100)
|
|
return candidates[-1]
|
|
|
|
if prev_direction == 'right':
|
|
if prev_event.type == 'plays' and len(pref.preffered_plays.all()) < 2:
|
|
candidates = nearest_plays(prev_event, 2)
|
|
return candidates[1]
|
|
if prev_event.type == 'movie' and len(pref.preffered_movies.all()) < 2:
|
|
candidates = nearest_movie(prev_event, 2)
|
|
return candidates[1]
|
|
if prev_event.type == 'concert' and len(pref.preferred_concerts.all()) < 2:
|
|
candidates = nearest_concert(prev_event, 2)
|
|
return candidates[1]
|
|
|
|
if prev_event.type == 'plays':
|
|
if not len(pref.preffered_movies.all()) and not len(pref.unpreffered_movies.all()):
|
|
return choice(Event.objects.filter(type='movie'))
|
|
if not len(pref.preferred_concerts.all()) and not len(pref.unpreferred_concerts.all()):
|
|
return choice(Event.objects.filter(type='concert'))
|
|
|
|
if prev_event.type == 'movie':
|
|
if not len(pref.preffered_plays.all()) and not len(pref.unpreffered_plays.all()):
|
|
return choice(Event.objects.filter(type='plays'))
|
|
if not len(pref.preferred_concerts.all()) and not len(pref.unpreferred_concerts.all()):
|
|
return choice(Event.objects.filter(type='concert'))
|
|
|
|
if prev_event.type == 'concert':
|
|
if not len(pref.preffered_plays.all()) and not len(pref.unpreffered_plays.all()):
|
|
return choice(Event.objects.filter(type='plays'))
|
|
if not len(pref.preffered_movies.all()) and not len(pref.unpreffered_movies.all()):
|
|
return choice(Event.objects.filter(type='movie'))
|
|
|
|
return None
|
|
|
|
|
|
|
|
def rank_candidates(candidates_list, negative_candidates_list):
|
|
flatten_c_list = []
|
|
ranks = {}
|
|
|
|
flatten_negatives = []
|
|
|
|
for negative in negative_candidates_list:
|
|
flatten_negatives.extend(negative)
|
|
|
|
for lst in candidates_list:
|
|
flatten_c_list.extend(lst)
|
|
for cand in lst:
|
|
ranks.update({cand: {'rank': 0, 'lst': lst}})
|
|
|
|
cnt = Counter(flatten_c_list)
|
|
|
|
for candidate, how_many in cnt.most_common(len(flatten_c_list)):
|
|
ranks[candidate]['rank'] = how_many * (len(ranks[candidate]['lst']) - ranks[candidate]['lst'].index(candidate))
|
|
|
|
res = []
|
|
for cand in ranks.keys():
|
|
res.append((ranks[cand]['rank'], cand))
|
|
return list(filter(lambda x: x[1] not in flatten_negatives, sorted(res, key=lambda x: -x[0])))
|
|
|
|
|
|
def get_personal_recommendation(prefer, unprefer):
|
|
candidates = []
|
|
negative_candidates = []
|
|
|
|
for rec in prefer:
|
|
candidates.append(list(map(lambda x: x.oid, get_nearest_event(rec, 10)[1:])))
|
|
|
|
for neg in unprefer:
|
|
negative_candidates.append(list(map(lambda x: x.oid, get_nearest_event(neg, 10)[1:])))
|
|
|
|
ranked = rank_candidates(candidates, negative_candidates)
|
|
|
|
return list(map(lambda x: (x[0], Event.objects.get(oid=x[1])), ranked[0:5]))
|
|
|
|
|
|
def get_personal_plays_recommendation(user):
|
|
pref = UserPreferences.objects.get(user=user)
|
|
|
|
prefer = pref.preffered_plays.all()
|
|
unprefer = pref.unpreffered_plays.all()
|
|
return get_personal_recommendation(prefer, unprefer)
|
|
|
|
|
|
def get_personal_concerts_recommendation(user):
|
|
pref = UserPreferences.objects.get(user=user)
|
|
|
|
prefer = pref.preferred_concerts.all()
|
|
unprefer = pref.unpreferred_concerts.all()
|
|
return get_personal_recommendation(prefer, unprefer)
|
|
|
|
|
|
def get_personal_movies_recommendation(user):
|
|
pref = UserPreferences.objects.get(user=user)
|
|
|
|
prefer = pref.preffered_movies.all()
|
|
unprefer = pref.unpreffered_movies.all()
|
|
return get_personal_recommendation(prefer, unprefer)
|