llmengine/base_embedding.py
2025-06-19 17:23:28 +08:00

12 lines
210 B
Python

class BaseEmbedding:
def embedding(self, doc):
es = self.model.encode([doc])
return es[0]
def similarity(self, qvector, dcovectors):
s = self.model.similarity([qvector], docvectors)
return s[0]