Class: Broadlistening::Steps::Embedding

Inherits:
BaseStep
  • Object
show all
Defined in:
lib/broadlistening/steps/embedding.rb

Constant Summary collapse

BATCH_SIZE = 
1000

Instance Attribute Summary

Attributes inherited from BaseStep

#config, #context

Instance Method Summary collapse

Methods inherited from BaseStep

#initialize

Constructor Details

This class inherits a constructor from Broadlistening::Steps::BaseStep

Instance Method Details

#execute ⇒ Object 



8
9
10
11
12
13
14
15
 
# File 'lib/broadlistening/steps/embedding.rb', line 8

def execute
  return context if context.arguments.empty?

  embeddings = compute_embeddings(context.arguments)
  attach_embeddings_to_arguments(context.arguments, embeddings)

  context
end