From 53d0b162d1630771002b58200966c999ddb957f1 Mon Sep 17 00:00:00 2001
From: Gea-Suan Lin <gslin@gslin.org>
Date: Wed, 28 Feb 2024 06:06:38 +0800
Subject: [PATCH] Avoid from getting article number repeatly.

---
 cmd/ir-tfidf/main.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cmd/ir-tfidf/main.go b/cmd/ir-tfidf/main.go
index fa9dde3..032a921 100644
--- a/cmd/ir-tfidf/main.go
+++ b/cmd/ir-tfidf/main.go
@@ -42,6 +42,7 @@ func main() {
 
 	fmt.Println("TF & DF Built")
 
+	n := len(articles)
 	q := strings.ToLower(os.Args[1])
 	q_tokens := tokenizer.Tokenize(q)
 
@@ -49,7 +50,7 @@ func main() {
 		var score float64 = 0.0
 		for _, w := range q_tokens {
 			if tf[w] != nil {
-				score += float64(tf[w][article.Id]) * math.Log2(float64(len(articles) / df[w]))
+				score += float64(tf[w][article.Id]) * math.Log2(float64(n / df[w]))
 			}
 		}