summaryrefslogtreecommitdiff
path: root/core/scoring/idf.go
diff options
context:
space:
mode:
Diffstat (limited to 'core/scoring/idf.go')
-rw-r--r--core/scoring/idf.go17
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
+}