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

Add related problems

Browse source
This commit is contained in:
cuom1999 2023-01-27 19:15:37 -06:00
parent 03455fca2c
commit ca13ee4e8d
7 changed files with 116 additions and 84 deletions
Download patch file Download diff file Expand all files Collapse all files

6
judge/ml/collab_filter.py
View file

@ -65,7 +65,7 @@ class CollabFilter:
return res
# return a list of pid
def problems_neighbors(self, problem, problemset, measure=DOT, limit=None):
def problem_neighbors(self, problem, problemset, measure=DOT, limit=None):
pid = problem.id
if pid >= len(self.problem_embeddings):
return None
@ -74,7 +74,7 @@ class CollabFilter:
)
res = []
for p in problemset:
if p.id < len(scores):
res.append((scores[p.id], p))
if p < len(scores):
res.append((scores[p], p))
res.sort(reverse=True, key=lambda x: x[0])
return res[:limit]