帮我利用pytorch和Bert写一个情感分类的代码,要求输入是一组句子对,输出格式为numpy
时间: 2023-04-05 19:00:33 浏览: 116
好的,我可以回答这个问题。以下是一个基于PyTorch和Bert的情感分类代码,输入为一组句子对,输出格式为numpy:
```
import torch
from transformers import BertTokenizer, BertForSequenceClassification
# 加载Bert模型和tokenizer
tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
model = BertForSequenceClassification.from_pretrained('bert-base-chinese', num_labels=2)
# 定义情感分类函数
def predict_sentiment(sentence_pair):
# 对句子进行tokenize和padding
inputs = tokenizer(sentence_pair, padding=True, truncation=True, return_tensors='pt')
# 使用Bert模型进行预测
outputs = model(**inputs)
# 获取预测结果
logits = outputs.logits
# 将结果转换为numpy格式
return logits.detach().numpy()
# 示例输入
sentence_pair = ["这个电影真的很好看", "这个电影真的很难看"]
# 进行情感分类预测
result = predict_sentiment(sentence_pair)
print(result)
```
输出结果为一个numpy数组,其中每个元素表示对应句子对的情感分类结果。
阅读全文