该模型在评估集上取得了以下结果:
本模型基于CamemBERT进行微调,专为文本情感分类这一特定任务而设计。其目标是根据训练数据集,将文本输入分类为不同的情感类别,如喜悦、悲伤、愤怒等。微调过程优化了模型,使其能更好地理解和处理法语文本。
本模型旨在用于涉及法语文本情感分类的应用场景。适用的应用包括:
尽管该模型经过训练,能在各类文本上表现良好,但仍存在一些局限性:
该模型基于 CamemBERT 架构,而 CamemBERT 本身是 BERT 模型的一个变体,专为法语文本优化。CamemBERT 是一种基于 transformer 的模型,在一系列法语自然语言处理任务中展现出了最先进的性能。
import torch
from openmind import AutoModelForSequenceClassification, AutoTokenizer
from openmind_hub import snapshot_download
import torch_npu
device = torch.device('npu')
import argparse
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--model_name_or_path",
type=str,
help="model_path",
default="Jinan_AICC/french_emotion_camembert",
)
args = parser.parse_args()
return args
args = parse_args()
model_name = args.model_name_or_path
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
model.to(device)
text = "Je suis très heureux de votre service rapide et efficace."
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
inputs = {key: value.to(device) for key, value in inputs.items()}
with torch.no_grad():
outputs = model(**inputs)
prediction = torch.nn.functional.softmax(outputs.logits, dim=-1)
predicted_emotion = prediction.argmax().item()
predicted_emotion = predicted_emotion
print("Predicted emotion:", predicted_emotion)