Change TF from [id][term] to [term][id].
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user