11 lines
596 B
YAML
11 lines
596 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: ["天气查询","互联网查询","页面切换","日计划数量查询","周计划数量查询","日计划作业内容","周计划作业内容",
|
|
"施工人数","作业考勤人数","知识问答","通用对话","作业面查询","班组人数查询","班组数查询","作业面内容",
|
|
"班组详情","工程进度查询"] # 类别名称
|