Class Embedder

• Defined in File cosmos.cpp

Class Documentation

class Embedder

Single CPU-core implementation of an Embedder/Similiarity scorer.

Turns sets of ‘

’ strings into parallel batches and either outputs embedding vectors or similarity scores.

Public Functions

Embedder(const std::string &modelPath, const std::string &vocabPath, bool computeSimilarity = false)

std::vector<std::vector<float>> embed(const std::string &input)

std::vector<float> similarity(const std::string &input1, const std::string &input2)