DCausse has uploaded a new change for review.
https://gerrit.wikimedia.org/r/272722
Change subject: cqd: add --docFilter
......................................................................
cqd: add --docFilter
Sometimes we may want to compare few docs that are not close in the ranking.
--docFilter will simply allow the user to pass a regex that will be applied
to the doc title, e.g.:
./cqd.py -w en.wikipedia.org -q "einstein" --allField=no -l 20
--docFilter='(^albert einstein$|refrigerator)'
Change-Id: I8ee5d574bb197bcdae5726ea620bdcfecc11b851
---
M cqd.py
1 file changed, 6 insertions(+), 1 deletion(-)
git pull ssh://gerrit.wikimedia.org:29418/wikimedia/discovery/relevancylab
refs/changes/22/272722/1
diff --git a/cqd.py b/cqd.py
index a14852e..073905f 100755
--- a/cqd.py
+++ b/cqd.py
@@ -314,6 +314,9 @@
self.indentChar = " "
self.printer = CQPrinter()
self.hitPrinter = CQHitPrinter(args=args)
+ self.docFilter = re.compile('.*')
+ if args.docFilter is not None:
+ self.docFilter = re.compile(args.docFilter, re.IGNORECASE)
def num(self, num):
self.printer.w(num, 'white')
@@ -348,7 +351,8 @@
self.printer.nl()
for h in results.hits:
- self.hitPrinter.disp(h, maxScore=results.max_score)
+ if self.docFilter.search(h.title):
+ self.hitPrinter.disp(h, maxScore=results.max_score)
class CQExplain:
@@ -820,6 +824,7 @@
aparser.add_argument('-pw', '--phraseWindow', type=int, help='Phrase window
size')
aparser.add_argument('-rp', '--rescoreProfile', help='Rescore profile')
aparser.add_argument('-disf', '--dismaxFilter', help='Filter DisMax fields to
display')
+aparser.add_argument('-docf', '--docFilter', help='Filter docs to display')
aparser.add_argument('-c', '--custom', nargs='+', default=[],
help='List of custom param (-c param1=value1
param2=value2)')
args = aparser.parse_args()
--
To view, visit https://gerrit.wikimedia.org/r/272722
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: I8ee5d574bb197bcdae5726ea620bdcfecc11b851
Gerrit-PatchSet: 1
Gerrit-Project: wikimedia/discovery/relevancylab
Gerrit-Branch: master
Gerrit-Owner: DCausse <[email protected]>
_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits
