Class: GTE::Model

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

Instance Method Summary collapse

Constructor Details

#initialize(dir, num_threads: 0, optimization_level: 3, model_name: nil) ⇒ Model

Returns a new instance of Model.



16
17
18
 
# File 'lib/gte.rb', line 16

def initialize(dir, num_threads: 0, optimization_level: 3, model_name: nil)
  @embedder = GTE::Embedder.new(dir, num_threads, optimization_level, model_name.to_s)
end

Instance Method Details

#[](input) ⇒ Object 



28
29
30
31
32
33
 
# File 'lib/gte.rb', line 28

def [](input)
  case input
  when String then embed(input).row(0)
  when Array  then embed(input)
  end
end

#embed(texts) ⇒ Object 



20
21
22
23
24
25
26
 
# File 'lib/gte.rb', line 20

def embed(texts)
  if texts.is_a?(String)
    @embedder.embed_one(texts)
  else
    @embedder.embed(Array(texts))
  end
end