update config

Signed-off-by: cwj <talkingwallace@sohu.com>
FederatedAI · Sep 13, 2023 · a120314 · a120314
1 parent 10821b2
commit a120314
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 20 deletions.
diff --git 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
@@ -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(