added field score calcualtions giving the title a higher weight

2 files changed, 11 insertions, 1 deletions

diff --git a/scoring/field.go b/scoring/field.go
new file mode 100644
index 0000000..9637950
--- /dev/null
+++ b/scoring/field.go
@@ -0,0 +1,10 @@
+package scoring
+
+import "searchEngine/index"
+
+var TitleWeight = 5.0
+var BodyWeight = 1.0
+
+func FieldScore(p index.Posting) float64 {
+	return (float64(p.TitleCount) * TitleWeight) + (float64(p.Count) * BodyWeight)
+}
diff --git a/scoring/idf.go b/scoring/idf.go
index 6d96089..5c85ba3 100644
--- a/scoring/idf.go
+++ b/scoring/idf.go
@@ -13,5 +13,5 @@ func CalculateIDF(totalDocs int, docFrequency int) float64 {
 }
 func ScoreTFIDF(p index.Posting, idf float64) float64 {
 
-	return float64(p.Count) * idf
+	return FieldScore(p) * idf
 }