From d72fe86325c1b0f8d3415ff9d58ea2a59a5c41d0 Mon Sep 17 00:00:00 2001 From: Gea-Suan Lin Date: Wed, 28 Feb 2024 21:54:15 +0800 Subject: [PATCH] Use `id` instead of `article.Id`. --- cmd/ir-tfidf/main.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cmd/ir-tfidf/main.go b/cmd/ir-tfidf/main.go index 032a921..616fdd9 100644 --- a/cmd/ir-tfidf/main.go +++ b/cmd/ir-tfidf/main.go @@ -23,17 +23,18 @@ func main() { df := make(map[string]int) for _, article := range articles { + id := article.Id str := strings.ToLower(article.Title + "\n" + article.Body) bag := tokenizer.Tokenize(str) - tokens[article.Id] = bag + tokens[id] = bag for _, w := range bag { // Handle TF: if _, ok := tf[w]; !ok { tf[w] = make(map[string]int) } - tf[w][article.Id] += strings.Count(str, w) + tf[w][id] += strings.Count(str, w) // Handle DF: df[w] += 1