modify ranking name slightly

boyter · Jun 14, 2020 · 8bc88d7 · 8bc88d7
1 parent b39c090
commit 8bc88d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -175,7 +175,7 @@ func main() {
 		&processor.Ranker,
 		"ranker",
 		"bm25",
-		"set ranking algorithm [tfidf, tfidfl, bm25]",
+		"set ranking algorithm [tfidf, tfidf2, bm25]",
 	)
 
 	// the below flags we want but are not enabled as yet

diff --git a/processor/result_ranker.go b/processor/result_ranker.go
@@ -20,7 +20,7 @@ func rankResults(corpusCount int, results []*fileJob) []*fileJob {
 	switch Ranker {
 	case "bm25":
 		results = rankResultsBM25(corpusCount, results, calculateDocumentFrequency(results))
-	case "tfidfl":
+	case "tfidf2":
 		results = rankResultsTFIDF(corpusCount, results, calculateDocumentFrequency(results), false)
 	default:
 		results = rankResultsTFIDF(corpusCount, results, calculateDocumentFrequency(results), true)