From f5ccba98ac63ca378a0aea2a0f5620f43a71be27 Mon Sep 17 00:00:00 2001 From: "Signed-off-by: wmqwxb" <13628607970@163.com> Date: Tue, 19 Dec 2023 14:24:10 +0800 Subject: [PATCH] update yaml Signed-off-by: wmqwxb <13628607970@163.com> --- static/download.yaml | 61 ++++++++++++++ static/hetero_secureboost.yaml | 132 ++++++++++++++++++++++++++++-- static/upload.yaml | 143 +++++++++++++++++++++++++++++++++ 3 files changed, 331 insertions(+), 5 deletions(-) create mode 100644 static/download.yaml create mode 100644 static/upload.yaml diff --git a/static/download.yaml b/static/download.yaml new file mode 100644 index 00000000..63fccb1a --- /dev/null +++ b/static/download.yaml @@ -0,0 +1,61 @@ +component: + name: download + description: '' + provider: fate_flow + version: 2.0.0 + labels: [] + roles: + - guest + - host + - local + parameters: + name: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + namespace: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + path: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + input_artifacts: + data: {} + model: {} + output_artifacts: + data: {} + model: {} + metric: + metric: + types: + - json_metric + optional: false + stages: + - default + roles: + - guest + - host + - local + description: metric, invisible for user + is_multi: false +schema_version: v1 diff --git a/static/hetero_secureboost.yaml b/static/hetero_secureboost.yaml index 0bd2ed6f..4d644ce0 100644 --- a/static/hetero_secureboost.yaml +++ b/static/hetero_secureboost.yaml @@ -1,5 +1,5 @@ component: - name: hetero_sbt + name: hetero_secureboost description: '' provider: fate version: 2.0.0-beta @@ -35,6 +35,16 @@ component: title: ConstrainedNumberMeta exclusiveMinimum: 0 type: integer + complete_secure: + type: ConstrainedNumberMeta + default: 0 + optional: true + description: number of trees to use guest features only in the complete secure + mode, 0 means no complete secure + type_meta: + title: ConstrainedNumberMeta + minimum: 0 + type: integer max_bin: type: ConstrainedNumberMeta default: 32 @@ -62,6 +72,34 @@ component: title: ConstrainedNumberMeta exclusiveMinimum: 0 type: integer + goss: + type: bool + default: false + optional: true + description: whether to use goss sampling + type_meta: + title: bool + type: boolean + default: true + description: whether output prediction result per cv fold + goss_start_iter: + type: ConstrainedNumberMeta + default: 0 + optional: true + description: start iteration of goss sampling + type_meta: + title: ConstrainedNumberMeta + exclusiveMinimum: 0 + type: integer + l1: + type: ConstrainedNumberMeta + default: 0 + optional: true + description: L1 regularization + type_meta: + title: ConstrainedNumberMeta + minimum: 0 + type: number l2: type: ConstrainedNumberMeta default: 0.1 @@ -69,7 +107,7 @@ component: description: L2 regularization type_meta: title: ConstrainedNumberMeta - exclusiveMinimum: 0 + minimum: 0 type: number min_impurity_split: type: ConstrainedNumberMeta @@ -116,7 +154,7 @@ component: title: bool type: boolean default: true - description: whether to use histogram subtraction + description: whether output prediction result per cv fold split_info_pack: type: bool default: true @@ -126,7 +164,7 @@ component: title: bool type: boolean default: true - description: whether to use histogram subtraction + description: whether output prediction result per cv fold hist_sub: type: bool default: true @@ -136,7 +174,7 @@ component: title: bool type: boolean default: true - description: whether to use histogram subtraction + description: whether output prediction result per cv fold he_param: type: HEParam default: @@ -167,6 +205,68 @@ component: key_length: 1024 description: homomorphic encryption param, support paillier, ou and mock in current version + top_rate: + type: ConstrainedNumberMeta + default: 0.2 + optional: true + description: top rate of goss sampling + type_meta: + title: ConstrainedNumberMeta + exclusiveMinimum: 0 + type: number + other_rate: + type: ConstrainedNumberMeta + default: 0.1 + optional: true + description: other rate of goss sampling + type_meta: + title: ConstrainedNumberMeta + exclusiveMinimum: 0 + type: number + cv_param: + type: CVParam + default: + n_splits: 5 + shuffle: false + random_state: + optional: true + description: cross validation param + type_meta: + title: CVParam + $ref: '#/definitions/abc__CVParam' + definitions: + abc__CVParam: + title: CVParam + type: object + properties: + n_splits: + title: N Splits + exclusiveMinimum: 1 + type: integer + shuffle: + title: Shuffle + default: false + type: boolean + random_state: + title: Random State + type: integer + required: + - n_splits + default: + n_splits: 5 + shuffle: false + random_state: + description: cross validation param + output_cv_data: + type: bool + default: true + optional: true + description: whether output prediction result per cv fold + type_meta: + title: bool + type: boolean + default: true + description: whether output prediction result per cv fold input_artifacts: data: train_data: @@ -202,6 +302,17 @@ component: - host description: '' is_multi: false + cv_data: + types: + - dataframe + optional: false + stages: + - cross_validation + roles: + - guest + - host + description: '' + is_multi: false model: train_model_input: types: @@ -249,6 +360,17 @@ component: - host description: '' is_multi: false + cv_output_datas: + types: + - dataframe + optional: true + stages: + - cross_validation + roles: + - guest + - host + description: '' + is_multi: true model: train_model_output: types: diff --git a/static/upload.yaml b/static/upload.yaml new file mode 100644 index 00000000..4027ef45 --- /dev/null +++ b/static/upload.yaml @@ -0,0 +1,143 @@ +component: + name: upload + description: '' + provider: fate_flow + version: 2.0.0 + labels: [] + roles: + - guest + - host + - local + parameters: + name: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + namespace: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + file: + type: str + default: + optional: false + description: '' + type_meta: + title: str + type: string + default: + description: '' + storage_engine: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + head: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + partitions: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + extend_sid: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + is_temp_file: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + address: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + meta: + type: str + default: + optional: true + description: '' + type_meta: + title: str + type: string + default: + description: '' + input_artifacts: + data: {} + model: {} + output_artifacts: + data: + table: + types: + - table + optional: false + stages: + - default + roles: + - guest + - host + - local + description: '' + is_multi: false + model: {} + metric: + metric: + types: + - json_metric + optional: false + stages: + - default + roles: + - guest + - host + - local + description: metric, invisible for user + is_multi: false +schema_version: v1