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

Fix bug

Browse source
This commit is contained in:
cuom1999 2023-01-28 02:20:53 -06:00
parent e0ce058989
commit c3f2930d4a
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
judge/ml/collab_filter.py
View file

@ -68,7 +68,7 @@ class CollabFilter:
def problem_neighbors(self, problem, problemset, measure=DOT, limit=None): def problem_neighbors(self, problem, problemset, measure=DOT, limit=None):
pid = problem.id pid = problem.id
if pid >= len(self.problem_embeddings): if pid >= len(self.problem_embeddings):
return None return []
scores = self.compute_scores( scores = self.compute_scores(
self.problem_embeddings[pid], self.problem_embeddings, measure self.problem_embeddings[pid], self.problem_embeddings, measure
) )