j1503 commited on
Commit
27dddc3
·
verified ·
1 Parent(s): 149d53a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -408,6 +408,9 @@ class CSCBM25Index(CSCInvertedIndex):
408
  indices.append(docid)
409
  indptr.append(len(data))
410
 
 
 
 
411
  posting_lists_matrix = csc_matrix(
412
  (data, indices, indptr),
413
  shape=(total_docs, len(posting_lists))
@@ -490,6 +493,7 @@ class BaseCSCInvertedIndexRetriever(BaseRetriever):
490
  continue
491
  tid = self.index.vocab[tok]
492
  weight = self.index.posting_lists_matrix[target_docid, tid]
 
493
  term_weights[tok] = weight
494
  return term_weights
495
  ## YOUR_CODE_ENDS_HERE
 
408
  indices.append(docid)
409
  indptr.append(len(data))
410
 
411
+ data = np.array(data, dtype=np.float32)
412
+ indices = np.array(indices, dtype=np.int32)
413
+ indptr = np.array(indptr, dtype=np.int32)
414
  posting_lists_matrix = csc_matrix(
415
  (data, indices, indptr),
416
  shape=(total_docs, len(posting_lists))
 
493
  continue
494
  tid = self.index.vocab[tok]
495
  weight = self.index.posting_lists_matrix[target_docid, tid]
496
+ if weight == 0: continue
497
  term_weights[tok] = weight
498
  return term_weights
499
  ## YOUR_CODE_ENDS_HERE