fusion

type Config ¶

type Config struct {
	Strategy string             // "rrf", "weighted", "max"
	K        int                // RRF parameter
	Weights  map[string]float64 // Strategy weights
}

Config holds configuration for fusion strategies

type Fusion ¶

type Fusion interface {
	// Fuse combines results from multiple strategies into a single ranked list
	// strategyResults maps strategy name to its results
	Fuse(strategyResults map[string][]database.SearchResult) ([]database.SearchResult, error)
}

Fusion defines the interface for combining results from multiple retrieval strategies

func New(config Config) (Fusion, error)

type MaxScoreFusion ¶

type MaxScoreFusion struct{}

MaxScoreFusion combines results by taking the maximum score for each document Useful when strategies use the same scoring scale

func NewMaxScoreFusion() *MaxScoreFusion

func (mf *MaxScoreFusion) Fuse(strategyResults map[string][]database.SearchResult) ([]database.SearchResult, error)

type ReciprocalRankFusion ¶

type ReciprocalRankFusion struct {
	// contains filtered or unexported fields
}

ReciprocalRankFusion implements the RRF (Reciprocal Rank Fusion) algorithm RRF is strategy-agnostic and works well for combining results from different retrieval methods

Formula: score(d) = Σ(1 / (k + rank(d))) where k is typically 60, and rank starts at 1 for the top result

Reference: "Reciprocal Rank Fusion outperforms Condorcet and individual Rank Learning Methods" by Cormack, Clarke, and Buettcher (SIGIR 2009)

func NewReciprocalRankFusion(k int) *ReciprocalRankFusion

func (rrf *ReciprocalRankFusion) Fuse(strategyResults map[string][]database.SearchResult) ([]database.SearchResult, error)

type WeightedFusion ¶

type WeightedFusion struct {
	// contains filtered or unexported fields
}

WeightedFusion combines results using weighted sum of strategy scores Each strategy's score is multiplied by its weight

func NewWeightedFusion(weights map[string]float64) *WeightedFusion

func (wf *WeightedFusion) Fuse(strategyResults map[string][]database.SearchResult) ([]database.SearchResult, error)