FATE icon indicating copy to clipboard operation
FATE copied to clipboard

验证阶段读不到模型的输出?

Open FancyXun opened this issue 1 year ago • 11 comments
trafficstars

YAML文件如下:

dag:
  parties:
  - party_id: ['9999']
    role: guest
  - party_id: ['10000']
    role: host
  party_tasks:
    guest_9999:
      parties:
      - party_id: ['9999']
        role: guest
      tasks:
        reader_0:
          parameters: {name: breast_hetero_guest, namespace: experiment}
    host_10000:
      parties:
      - party_id: ['10000']
        role: host
      tasks:
        reader_0:
          parameters: {name: breast_hetero_host, namespace: experiment}
  stage: train
  tasks:
    eval_0:
      component_ref: evaluation
      dependent_tasks: [sbt_0]
      inputs:
        data:
          input_data:
            task_output_artifact:
            - output_artifact_key: train_data_output
              parties:
              - party_id: ['9999']
                role: guest
              producer_task: sbt_0
      parameters:
        label_column_name: null
        metrics: [auc]
        predict_column_name: null
      parties:
      - party_id: ['9999']
        role: guest
      stage: default

    psi_0:
      component_ref: psi
      dependent_tasks: [reader_0]
      inputs:
        data:
          input_data:
            task_output_artifact:
              output_artifact_key: output_data
              parties:
              - party_id: ['9999']
                role: guest
              - party_id: ['10000']
                role: host
              producer_task: reader_0
      parameters: {}
      stage: default
    reader_0:
      component_ref: reader
      parameters: {}
      stage: default
    sbt_0:
      component_ref: hetero_secureboost
      dependent_tasks: [psi_0]
      inputs:
        data:
          cv_data:
            task_output_artifact:
              output_artifact_key: output_data
              parties:
              - party_id: ['9999']
                role: guest
              - party_id: ['10000']
                role: host
              producer_task: psi_0
        model: {}
      parameters:
        cv_param: {n_splits: 3}
        gh_pack: true
        goss: false
        goss_start_iter: 0
        he_param: {key_length: 1024, kind: paillier}
        hist_sub: true
        l1: 0
        l2: 0.1
        learning_rate: 0.3
        max_bin: 32
        max_depth: 2
        min_child_weight: 1
        min_impurity_split: 0.01
        min_leaf_node: 1
        min_sample_split: 2
        num_class: 2
        num_trees: 2
        objective: binary:bce
        other_rate: 0.1
        split_info_pack: true
        top_rate: 0.2
      stage: cross_validation
schema_version: 2.0.0

提交后

Screen Shot 2024-07-11 at 16 06 48

报错如下:

[ERROR][2024-07-11 07:54:39,269][585110][_wraps.run][line:92]: Get data artifacts failed: {'job_id': '202407110739162543270', 'role': 'guest', 'party_id': '9999', 'task_name': 'sbt_0', 'output_key': 'train_data_output'}, response: {"code":2005,"message":"failed"}
[ERROR][2024-07-11 07:54:39,269][585110][_wraps.run][line:92]: Get data artifacts failed: {'job_id': '202407110739162543270', 'role': 'guest', 'party_id': '9999', 'task_name': 'sbt_0', 'output_key': 'train_data_output'}, response: {"code":2005,"message":"failed"}

感觉像是sbt没有输出模型吗?所以fate-server没找到?这个yaml文件我也是基于 https://github.com/FederatedAI/FATE/blob/v2.0.0/examples/pipeline/hetero_secureboost/test_hetero_sbt_binary.py 这个生成的

FancyXun avatar Jul 11 '24 07:07 FancyXun