From d6e1c1dbf56a72729208ec206428b1659cfe0a91 Mon Sep 17 00:00:00 2001 From: Gea-Suan Lin Date: Fri, 9 Feb 2024 15:16:47 +0800 Subject: [PATCH] Implement query to tf-idf score. --- cmd/ir-tfidf/main.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/cmd/ir-tfidf/main.go b/cmd/ir-tfidf/main.go index 8899b8e..f4f42a2 100644 --- a/cmd/ir-tfidf/main.go +++ b/cmd/ir-tfidf/main.go @@ -2,6 +2,8 @@ package main import ( "fmt" + "math" + "os" "strings" "github.com/gslin/go-ir-playground/internal/artifact" @@ -34,4 +36,16 @@ func main() { } fmt.Println("TF & DF Built") + + q := os.Args[1:] + + for _, article := range articles { + var score float64 = 0.0 + for _, w := range q { + if tf[w] != nil { + score += float64(tf[w][article.Id]) * math.Log2(float64(len(articles) / df[w])) + } + } + fmt.Printf("Article %v: %v\n", article.Id, score) + } }