lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix sort key error

Browse source
This commit is contained in:
cuom1999 2022-04-12 11:53:39 -05:00
parent e9725d27aa
commit 5c6391fb76
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
judge/ml/collab_filter.py
View file

@ -45,7 +45,7 @@ class CollabFilter:
if pid < len(scores):
res.append((scores[pid], problem))
res.sort(reverse=True)
res.sort(reverse=True, key=lambda x: x[0])
return res[:limit]
@ -60,5 +60,5 @@ class CollabFilter:
for p in problemset:
if p.id < len(scores):
res.append((scores[p.id], p))
res.sort(reverse=True)
res.sort(reverse=True, key=lambda x: x[0])
return res[:limit]

3
judge/views/problem.py
View file

@ -646,7 +646,8 @@ class ProblemFeed(FeedView):
cl_model = CollabFilter()
dot_rec = cl_model.user_recommendations(user, queryset, cl_model.DOT, 100)
cosine_rec = cl_model.user_recommendations(user, queryset, cl_model.COSINE, 100)
hot_problems_rec = hot_problems(timedelta(days=7), 10)
hot_problems_rec = [problem for problem in hot_problems(timedelta(days=7), 10)
if problem in queryset]
q = self.merge_recommendation([dot_rec, cosine_rec, hot_problems_rec])
return q