mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2025-03-07 15:08:51 +08:00
Code Issues Packages Projects Releases Wiki Activity
73ba312cf8
cube-studio/job-template/job/xgb
History
data-infra 77d3418e18 升级任务模板
2024-02-28 14:56:05 +08:00
..
build.sh 去除framworker依赖 2023-12-10 20:01:38 +08:00
Dockerfile 升级任务模板 2024-02-28 14:56:05 +08:00
launcher.py 升级任务模板 2024-02-28 14:56:05 +08:00
README.md 去除非必要模板参数说明 2023-12-16 20:47:24 +08:00

README.md

xgboost 模板

描述：单机xgb训练，支持训练预测。

镜像：ccr.ccs.tencentyun.com/cube-studio/xgb:20230801

启动参数：

{
    "训练推理": {
      "--train_dataset": {
        "type": "str",
        "item_type": "str",
        "label": "训练数据集",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "训练数据集",
        "editable": 1
      },
      "--val_dataset": {
        "type": "str",
        "item_type": "str",
        "label": "评估数据集",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "评估数据集",
        "editable": 1
      },
      "--feature_columns": {
        "type": "str",
        "item_type": "str",
        "label": "特征列，逗号分隔",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "特征列，逗号分隔",
        "editable": 1
      },
      "--label_columns": {
        "type": "str",
        "item_type": "str",
        "label": "标签列",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "标签列，逗号分割",
        "editable": 1
      },
      "--model_params": {
        "type": "json",
        "item_type": "str",
        "label": "模型参数",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "模型参数",
        "editable": 1
      },
      "--save_model_dir": {
        "type": "str",
        "item_type": "str",
        "label": "模型保存目录",
        "require": 1,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "模型保存目录",
        "editable": 1
      },
      "--inference_dataset": {
        "type": "str",
        "item_type": "str",
        "label": "推理数据集",
        "require": 0,
        "choice": [],
        "range": "",
        "default": "",
        "placeholder": "",
        "describe": "推理数据集",
        "editable": 1
      }
    }
}