summaryrefslogtreecommitdiff
path: root/core/scoring/idf.go
blob: f5aaecd5d66d5033dc04a38612df80a0ce599d97 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
package scoring

import (
	"math"
	"searchEngine/core/index"
)

func CalculateIDF(totalDocs int, docFrequency int) float64 {
	if docFrequency == 0 {
		return 0.0
	}
	return math.Log10(float64(totalDocs) / float64(docFrequency))
}

func ScoreTFIDF(p index.Posting, idf float64) float64 {
	return FieldScore(p) * idf
}