对pandas数据框架的所有值应用Bert编码

Question

我正在尝试获得数据框架中存在的所有值的bert嵌入。

我的代码如下:

from sentence_transformers import SentenceTransformer, util
model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
import pandas as pd
sentences =[ ["I'm happy", "I'm full of happiness"], ["I am sam", "I am good"]]
df = pd.DataFrame(sentences)
encoded_df = df.applymap(model.encode(convert_to_tensor=True))

面对以下错误:

encode() missing 1 required positional argument: 'sentences'

有什么方法可以做到这一点吗?

Answer 1

可以使用

encoded_df = df.applymap(lambda x: model.encode(x, convert_to_tensor=True))