From a12031464ebc859a8778176d172b3a494c7383ae Mon Sep 17 00:00:00 2001
From: cwj <talkingwallace@sohu.com>
Date: Wed, 13 Sep 2023 10:54:54 +0800
Subject: [PATCH] update config

Signed-off-by: cwj <talkingwallace@sohu.com>
---
 .../hetero_sbt/hetero_sbt_benchmark.yaml      | 36 +++++++++----------
 .../hetero_sbt/pipeline_hetero_sbt.py         |  4 +--
 2 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/examples/benchmark_quality/hetero_sbt/hetero_sbt_benchmark.yaml b/examples/benchmark_quality/hetero_sbt/hetero_sbt_benchmark.yaml
index 428846b828..a84fdddf4c 100644
--- a/examples/benchmark_quality/hetero_sbt/hetero_sbt_benchmark.yaml
+++ b/examples/benchmark_quality/hetero_sbt/hetero_sbt_benchmark.yaml
@@ -101,15 +101,15 @@ data:
     table_name: student_hetero_host
     namespace: experiment
     role: host_0  
-# hetero_sbt-binary-0-breast:
-#   local:
-#     script: "./xgb.py"
-#     conf: "./xgb_breast_config.yaml"
-#   FATE-hetero-sbt:
-#     script: "./pipeline_hetero_sbt.py"
-#     conf: "./sbt_breast_config.yaml"
-#   compare_setting:
-#     relative_tol: 0.01
+hetero_sbt-binary-0-breast:
+  local:
+    script: "./xgb.py"
+    conf: "./xgb_breast_config.yaml"
+  FATE-hetero-sbt:
+    script: "./pipeline_hetero_sbt.py"
+    conf: "./sbt_breast_config.yaml"
+  compare_setting:
+    relative_tol: 0.01
 hetero_sbt-regression:
   local:
     script: "./xgb.py"
@@ -119,14 +119,14 @@ hetero_sbt-regression:
     conf: "./sbt_student_config.yaml"
   compare_setting:
     relative_tol: 0.05
-# hetero_sbt-binary-1-default-credit:
-#   local:
-#     script: "./xgb.py"
-#     conf: "./xgb_default_credit_config.yaml"
-#   FATE-hetero-sbt:
-#     script: "./pipeline_hetero_sbt.py"
-#     conf: "./sbt_default_credit_config.yaml"
-#   compare_setting:
-#     relative_tol: 0.01
+hetero_sbt-binary-1-default-credit:
+  local:
+    script: "./xgb.py"
+    conf: "./xgb_default_credit_config.yaml"
+  FATE-hetero-sbt:
+    script: "./pipeline_hetero_sbt.py"
+    conf: "./sbt_default_credit_config.yaml"
+  compare_setting:
+    relative_tol: 0.01
 
 
diff --git a/examples/benchmark_quality/hetero_sbt/pipeline_hetero_sbt.py b/examples/benchmark_quality/hetero_sbt/pipeline_hetero_sbt.py
index 5a19e957d5..50ae538970 100644
--- a/examples/benchmark_quality/hetero_sbt/pipeline_hetero_sbt.py
+++ b/examples/benchmark_quality/hetero_sbt/pipeline_hetero_sbt.py
@@ -61,8 +61,8 @@ def main(config="../../config.yaml", param="./sbt_breast_config.yaml", namespace
         "objective": param.get("objective", "binary:bce"),
     }
     hetero_sbt_0 = HeteroSBT('sbt_0', train_data=psi_0.outputs['output_data'], num_trees=config_param["num_trees"], 
-                             max_bin=config_param["max_bin"], max_depth=config_param["max_depth"], he_param={'kind': 'paillier', 'key_length': 2048},
-                             objective=config_param["objective"], hist_sub=False, gh_pack=False, split_info_pack=False)
+                             max_bin=config_param["max_bin"], max_depth=config_param["max_depth"], he_param={'kind': 'ou', 'key_length': 1024},
+                             objective=config_param["objective"])
 
     if config_param['objective'] == 'regression:l2':
         evaluation_0 = Evaluation(