query_intent_detection_mapper

Predicts the user’s intent label and corresponding score for a given query. The operator uses a Hugging Face model to classify the intent of the input query. If the query is in Chinese, it can optionally be translated to English using another Hugging Face translation model before classification. The predicted intent label and its confidence score are stored in the meta field with the keys ‘query_intent_label’ and ‘query_intent_score’, respectively. If these keys already exist in the meta field, the operator will skip processing for those samples.

预测给定查询的用户意图标签及其对应分数。该算子使用 Hugging Face 模型对输入查询的意图进行分类。如果查询是中文，可以使用另一个 Hugging Face 翻译模型将其翻译成英文后再进行分类。预测的意图标签及其置信分数分别存储在 meta 字段中的 ‘query_intent_label’ 和 ‘query_intent_score’ 键下。如果这些键已经存在于 meta 字段中，算子将跳过这些样本的处理。

Type 算子类型: mapper

Tags 标签: cpu, hf, hf

🔧 Parameter Configuration 参数配置

name 参数名

type 类型

default 默认值

desc 说明

hf_model

<class ‘str’>

'bespin-global/klue-roberta-small-3i4k-intent-classification'

Huggingface model ID to predict intent 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_intent_label'

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

score_key

<class ‘str’>

'query_intent_label_score'

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

kwargs

''

Extra keyword arguments.

📊 Effect demonstration 效果演示

not available 暂无