9 lines
621 B
YAML
9 lines
621 B
YAML
model_path: E:\workingSpace\PycharmProjects\ernie\models\ernie-3.0-tiny-base-v2-zh #模型地址
|
|
train: E:\workingSpace\PycharmProjects\ernie\ernie\data\train.json # 训练集路径
|
|
val: E:\workingSpace\PycharmProjects\ernie\ernie\data\val.json # 验证集路径
|
|
test: E:\workingSpace\PycharmProjects\ernie\ernie\data\test.json # (可选) 测试集路径
|
|
|
|
# 类别信息
|
|
nc: 10 # 目标类别数
|
|
labels: ["天气查询","互联网查询","页面切换","日计划数量查询","周计划数量查询","日计划作业内容","周计划作业内容","施工人数","作业考勤人数","知识问答"] # 类别名称
|