Sean MacAvaney commited on
Commit
b036a52
1 Parent(s): a8fb57f
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -91,7 +91,8 @@ from pyterrier_dr import ElectraScorer
91
 
92
  doc2query = Doc2Query({repr(model)}, append=False, num_samples={num_samples})
93
  scorer = ElectraScorer({repr(score_model)})
94
- pipeline = doc2query >> QueryScorer(scorer) >> QueryFilter(t={query_filter.t})
 
95
 
96
  pipeline({df2code(input)})
97
  '''
 
91
 
92
  doc2query = Doc2Query({repr(model)}, append=False, num_samples={num_samples})
93
  scorer = ElectraScorer({repr(score_model)})
94
+ pipeline = doc2query >> QueryScorer(scorer) >> QueryFilter(append=False, t={query_filter.t})
95
+ # use append=True when indexing; t={query_filter.t} is the {filter_pct}th percentile for generated queries on MS MARCO
96
 
97
  pipeline({df2code(input)})
98
  '''