diff options
Diffstat (limited to 'core/scoring/idf.go')
| -rw-r--r-- | core/scoring/idf.go | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/core/scoring/idf.go b/core/scoring/idf.go new file mode 100644 index 0000000..f5aaecd --- /dev/null +++ b/core/scoring/idf.go @@ -0,0 +1,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 +} |
