Change TF from [id][term] to [term][id].

This commit is contained in:
Gea-Suan Lin
2024-02-09 14:33:07 +08:00
parent ade2049093
commit 2d447ad45b

View File

@@ -21,12 +21,12 @@ func main() {
bag := tokenizer.Tokenize(str) bag := tokenizer.Tokenize(str)
tokens[article.Id] = bag tokens[article.Id] = bag
// Init TF:
tf[article.Id] = make(map[string]int)
for _, w := range bag { for _, w := range bag {
// Handle TF: // Handle TF:
tf[article.Id][w] = strings.Count(str, w) if _, ok := tf[w]; !ok {
tf[w] = make(map[string]int)
}
tf[w][article.Id] += strings.Count(str, w)
// Handle DF: // Handle DF:
df[w] += 1 df[w] += 1