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]