diff options
Diffstat (limited to 'scoring')
| -rw-r--r-- | scoring/field.go | 10 | ||||
| -rw-r--r-- | scoring/idf.go | 2 |
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 } |