Class: GTE::Reranker

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

Class Method Summary collapse

Class Method Details

.native_new ⇒ Object 



6
 
# File 'lib/gte/reranker.rb', line 6

alias native_new new

.new(model_dir, &block) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
# File 'lib/gte/reranker.rb', line 8

def new(model_dir, &block)
  cfg = default_config(model_dir)
  cfg = block.call(cfg) if block
  native_new(
    cfg.model_dir,
    cfg.optimization_level,
    cfg.model_name.to_s,
    cfg.sigmoid,
    cfg.output_tensor.to_s,
    cfg.max_length || 0,
    cfg.padding.to_s,
    cfg.execution_providers.to_s
  )
end