Bert Large Uncased Sst2

🚀 用于情感分析的BERT-Large-Uncased模型

本模型是 bert-large-uncased 的微调版本。bert-large-uncased 最初发布于论文 "BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding"，并在 斯坦福情感树库v2 (SST2) 上进行训练，该数据集是 通用语言理解评估 (GLUE) 基准的一部分。此模型由 AssemblyAI 团队进行微调，并随 相应的博客文章 一同发布。

🚀 快速开始

💻 使用示例

基础用法

import torch.nn.functional as F 
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("assemblyai/bert-large-uncased-sst2") 
model = AutoModelForSequenceClassification.from_pretrained("assemblyai/bert-large-uncased-sst2")

tokenized_segments = tokenizer(["AssemblyAI is the best speech-to-text API for modern developers with performance being second to none!"], return_tensors="pt", padding=True, truncation=True)
tokenized_segments_input_ids, tokenized_segments_attention_mask = tokenized_segments.input_ids, tokenized_segments.attention_mask
model_predictions = F.softmax(model(input_ids=tokenized_segments_input_ids, attention_mask=tokenized_segments_attention_mask)['logits'], dim=1)

print("Positive probability: "+str(model_predictions[0][1].item()*100)+"%")
print("Negative probability: "+str(model_predictions[0][0].item()*100)+"%")

若您在使用此模型时遇到问题，欢迎随时联系 AssemblyAI 团队！