slight update

boyter · Jun 21, 2020 · 46b183d · 46b183d
1 parent 01c00ef
commit 46b183d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -182,7 +182,7 @@ func main() {
 		&processor.Ranker,
 		"ranker",
 		"bm25",
-		"set ranking algorithm [count, tfidf, tfidf2, bm25]",
+		"set ranking algorithm [wc, tfidf, tfidf2, bm25]",
 	)
 	flags.StringVarP(
 		&processor.FileOutput,

diff --git a/processor/result_ranker.go b/processor/result_ranker.go
@@ -18,7 +18,7 @@ import (
 func rankResults(corpusCount int, results []*fileJob) []*fileJob {
 	// needs to come first because it resets the scores
 	switch Ranker {
-	case "count":
+	case "wc":
 		results = results
 	case "bm25":
 		results = rankResultsBM25(corpusCount, results, calculateDocumentFrequency(results))