12 lines
210 B
Python
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]
|
|
|