推理意图
模块 ID: infer_email_intent
模块命名空间: morpheus_spear_phishing
推断给定电子邮件正文的“意图”。
参数 |
类型 |
描述 |
示例值 |
默认值 |
---|---|---|---|---|
intent |
string | 模型的意图 | "classify" |
None |
task |
string | 模型的任务 | "text-classification" |
"text-classification" |
model_path |
string | 模型的路径 | "/path/to/model" |
None |
truncation |
boolean | 如果为 true,则将输入截断为 max_length |
true |
true |
max_length |
integer | 模型输入的最大长度 | 512 |
512 |
batch_size |
integer | 处理批次的大小 | 256 |
256 |
feature_col |
string | 要使用的特征列 | "body" |
"body" |
label_col |
string | 要使用的标签列 | "label" |
"label" |
device |
integer | 运行设备 | 0 |
0 |
raise_on_failure |
boolean | 如果为 true,则在失败时引发异常 | false |
false |
{
"intent": "classify",
"task": "text-classification",
"model_path": "/path/to/model",
"truncation": true,
"max_length": 512,
"batch_size": 256,
"feature_col": "body",
"label_col": "label",
"device": 0,
"raise_on_failure": false
}