【发布时间】:2022-12-13 23:47:19
【问题描述】:
我想使用多语言 BERT 翻译我的数据框。
我已经复制了这段代码,但我想使用我自己的数据框代替text。
from transformers import BertTokenizer, TFBertModel
tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
model = TFBertModel.from_pretrained("bert-base-multilingual-cased")
text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='tf')
output = model(encoded_input)
但是,我在使用它时遇到一些错误,如下所示。
df =pd.read_csv("/content/drive/text.csv")
encoded_input = tokenizer(df, return_tensors='tf')
错误
ValueError: text input must of type `str` (single example), `List[str]` (batch or single pretokenized example) or `List[List[str]]` (batch of pretokenized examples).
我的数据框看起来像这样
0 There is XXXX increased opacity within the rig...
1 There is XXXX increased opacity within the rig...
2 There is XXXX increased opacity within the rig...
3 Interstitial markings are diffusely prominent ...
4 Interstitial markings are diffusely prominent ...
Name: findings, dtype: object
【问题讨论】:
标签: pandas nlp bert-language-model