gslin/go-ir-playground
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use id instead of article.Id.

Browse Source
This commit is contained in:
Gea-Suan Lin
2024-02-28 21:54:15 +08:00
parent 7290880a10
commit d72fe86325
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/ir-tfidf/main.go
View File

@@ -23,17 +23,18 @@ func main() {
df := make(map[string]int) df := make(map[string]int)
for _, article := range articles { for _, article := range articles {
id := article.Id
str := strings.ToLower(article.Title + "\n" + article.Body) str := strings.ToLower(article.Title + "\n" + article.Body)
bag := tokenizer.Tokenize(str) bag := tokenizer.Tokenize(str)
tokens[article.Id] = bag tokens[id] = bag
for _, w := range bag { for _, w := range bag {
// Handle TF: // Handle TF:
if _, ok := tf[w]; !ok { if _, ok := tf[w]; !ok {
tf[w] = make(map[string]int) tf[w] = make(map[string]int)
} }
tf[w][article.Id] += strings.Count(str, w) tf[w][id] += strings.Count(str, w)
// Handle DF: // Handle DF:
df[w] += 1 df[w] += 1