Class: GTE::Reranker

Inherits:
Object
  • Object
show all
Defined in:
lib/gte/reranker.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.config(model_dir) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
 
# File 'lib/gte/reranker.rb', line 6

def config(model_dir)
  cfg = default_config(model_dir)

  if block_given?
    yielded = yield(cfg)
    cfg = yielded if yielded.is_a?(Config::Reranker)
  end

  build(cfg)
end

Instance Method Details

#rerank(query:, candidates:) ⇒ Object 



46
47
48
49
50
51
52
53
54
 
# File 'lib/gte/reranker.rb', line 46

def rerank(query:, candidates:)
  rows = Array(candidates).map(&:to_s)
  scores = score(query.to_s, rows)

  rows
    .each_with_index
    .map { |text, idx| { index: idx, score: scores[idx], text: text } }
    .sort_by { |row| -row[:score] }
end