querysentimentdetection_mapper
February 4, 2026 · View on GitHub
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 标签: gpu, 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. |