Cross-encoder re-ranking (Kamath Ch 7)

Joint (q, d) scoring with a STABLE descending re-order; ranking is 0-BASED.

Usage

morie_kamath_cross_encoder_rerank(q, docs, model, top_k = NULL)

Arguments

q

Query.

docs

Documents.

model

Scorer.

top_k

Optional cut-off.

Value

List with estimate, scores, ranking, reranked.

Examples

morie_kamath_cross_encoder_rerank("q", list("a", "b", "c"), function(q,
    d) c(a = 0.1, b = 0.9, c = 0.9)[[d]])
#> $estimate
#> [1] 0.9
#> 
#> $scores
#> [1] 0.1 0.9 0.9
#> 
#> $ranking
#> [1] 1 2 0
#> 
#> $reranked
#> $reranked[[1]]
#> [1] "b"
#> 
#> $reranked[[2]]
#> [1] "c"
#> 
#> $reranked[[3]]
#> [1] "a"
#> 
#> 
#> $n
#> [1] 3
#> 
#> $method
#> [1] "cross-encoder re-ranking (Kamath Ch 7)"
#>