Spaces:
Runtime error
Runtime error
Sean MacAvaney
commited on
Commit
·
b036a52
1
Parent(s):
a8fb57f
comment
Browse files
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 |
'''
|