query_sentiment_detection_mapper

Predicts user’s sentiment label (‘negative’, ‘neutral’, ‘positive’) in a query.

This mapper takes input from the specified query key and outputs the predicted sentiment label and its corresponding score. The results are stored in the Data-Juicer meta field under ‘query_sentiment_label’ and ‘query_sentiment_label_score’. It uses a Hugging Face model for sentiment detection. If a Chinese-to-English translation model is provided, it first translates the query from Chinese to English before performing sentiment analysis.

预测查询中的用户情感标签（’negative’, ‘neutral’, ‘positive’）。

此映射器从指定的查询键获取输入，并输出预测的情感标签及其对应的分数。结果存储在 Data-Juicer meta 字段下的 ‘query_sentiment_label’ 和 ‘query_sentiment_label_score’ 中。它使用 Hugging Face 模型进行情感检测。如果提供了中文到英文的翻译模型，它会先将查询从中文翻译成英文再进行情感分析。

Type 算子类型: mapper

Tags 标签: cpu, hf, hf

🔧 Parameter Configuration 参数配置

name 参数名

type 类型

default 默认值

desc 说明

hf_model

<class ‘str’>

'mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis'

Huggingface model ID to predict sentiment label.

zh_to_en_hf_model

typing.Optional[str]

'Helsinki-NLP/opus-mt-zh-en'

Translation model from Chinese to English. If not None, translate the query from Chinese to English.

model_params

typing.Dict

{}

model param for hf_model.

zh_to_en_model_params

typing.Dict

{}

model param for zh_to_hf_model.

label_key

<class ‘str’>

'query_sentiment_label'

The key name in the meta field to store the output label. It is ‘query_sentiment_label’ in default.

score_key

<class ‘str’>

'query_sentiment_label_score'

The key name in the meta field to store the corresponding label score. It is ‘query_sentiment_label_score’ in default.

kwargs

''

Extra keyword arguments.

📊 Effect demonstration 效果演示

not available 暂无