从变压器进口BertTokenizer



我正在尝试从拥抱人脸实现以下模型,但不完全确定如何向模型提供我需要传递的文本来进行分类。文件(https://huggingface.co/DaNLP/da-bert-tone-subjective-objective)没有显示如何传递查询。

from transformers import BertTokenizer, BertForSequenceClassification
model = BertForSequenceClassification.from_pretrained("DaNLP/da-bert-tone-subjective-objective")
tokenizer = BertTokenizer.from_pretrained("DaNLP/da-bert-tone-subjective-objective")

请提供进一步指导。

几周前,我还使用了HF的一个预训练模型。。。下面是我用来测试输入的代码。。。你可以尝试同样的方法,如果对你有效,那么最好。。。否则,您可以在查询中共享的同一页面上查看文档或预训练链接。。。

sent = "paste your input here"
inputs = tokenizer.encode(sent, return_tensors='pt', padding=True)
outputs = model(inputs)
print(outputs)
or
print(outputs[0])
or
print(outputs[0].detach().numpy())

最新更新