Avoid from getting article number repeatly.
This commit is contained in:
@@ -42,6 +42,7 @@ func main() {
|
|||||||
|
|
||||||
fmt.Println("TF & DF Built")
|
fmt.Println("TF & DF Built")
|
||||||
|
|
||||||
|
n := len(articles)
|
||||||
q := strings.ToLower(os.Args[1])
|
q := strings.ToLower(os.Args[1])
|
||||||
q_tokens := tokenizer.Tokenize(q)
|
q_tokens := tokenizer.Tokenize(q)
|
||||||
|
|
||||||
@@ -49,7 +50,7 @@ func main() {
|
|||||||
var score float64 = 0.0
|
var score float64 = 0.0
|
||||||
for _, w := range q_tokens {
|
for _, w := range q_tokens {
|
||||||
if tf[w] != nil {
|
if tf[w] != nil {
|
||||||
score += float64(tf[w][article.Id]) * math.Log2(float64(len(articles) / df[w]))
|
score += float64(tf[w][article.Id]) * math.Log2(float64(n / df[w]))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user