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

5
judge/views/problem.py
View file

@ -646,8 +646,9 @@ class ProblemFeed(FeedView):
cl_model = CollabFilter() cl_model = CollabFilter()
dot_rec = cl_model.user_recommendations(user, queryset, cl_model.DOT, 100) dot_rec = cl_model.user_recommendations(user, queryset, cl_model.DOT, 100)
cosine_rec = cl_model.user_recommendations(user, queryset, cl_model.COSINE, 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]) q = self.merge_recommendation([dot_rec, cosine_rec, hot_problems_rec])
return q return q