12 lines
803 B
YAML
12 lines
803 B
YAML
model_path: ernie-3.0-tiny-base-v2-zh #模型地址
|
|
train: ./data/train.json # 训练集路径
|
|
val: ./data/val.json # 验证集路径
|
|
test: ./data/test.json # (可选) 测试集路径
|
|
|
|
# 类别信息
|
|
nc: 13 # 目标类别数
|
|
labels: ["天气查询","互联网查询","页面切换","日计划数量查询","周计划数量查询","日计划作业内容","周计划作业内容",
|
|
"施工人数","作业考勤人数","知识问答","通用对话","作业面查询","班组人数查询","班组数查询","作业面内容",
|
|
"班组详情","工程进度查询", "人员查询", "分公司查询","工程数量查询","工程详情查询","项目部数量查询",
|
|
"建管单位数量查询","建管单位详情","分包单位数量查询","分包单位详情"] # 类别名称
|