Merge pull request FederatedAI#2335 from FederatedAI/feature-1.5.0-fi…

…x_fast_sbt_bug fix fast sbt param naming error
KAgBMan · Nov 18, 2020 · fb209da · fb209da
2 parents f7753ca + d761a23
commit fb209da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/fate_client/pipeline/param/boosting_param.py b/python/fate_client/pipeline/param/boosting_param.py
@@ -473,7 +473,7 @@ def __init__(self, tree_param: DecisionTreeParam = DecisionTreeParam(), task_typ
                  predict_param=PredictParam(), cv_param=CrossValidationParam(),
                  validation_freqs=None, early_stopping=None, use_missing=False, zero_as_missing=False,
                  complete_secure=False, tree_num_per_party=1, guest_depth=1, host_depth=1, work_mode='mix', metrics=None,
-                 subsample_random_seed=None, binning_error=consts.DEFAULT_RELATIVE_ERROR):
+                 subsample_random_seed=None, binning_error=consts.DEFAULT_RELATIVE_ERROR, sparse_optimization=False):
 
         """
         work_mode：
@@ -497,7 +497,7 @@ def __init__(self, tree_param: DecisionTreeParam = DecisionTreeParam(), task_typ
                                                          predict_param, cv_param, validation_freqs, early_stopping,
                                                          use_missing, zero_as_missing, complete_secure, metrics=metrics,
                                                          subsample_random_seed=subsample_random_seed,
-                                                         binning_error=binning_error)
+                                                         binning_error=binning_error, sparse_optimization=sparse_optimization)
 
         self.tree_num_per_party = tree_num_per_party
         self.guest_depth = guest_depth

diff --git a/python/federatedml/param/boosting_param.py b/python/federatedml/param/boosting_param.py
@@ -468,7 +468,7 @@ def __init__(self, tree_param: DecisionTreeParam = DecisionTreeParam(), task_typ
                  predict_param=PredictParam(), cv_param=CrossValidationParam(),
                  validation_freqs=None, early_stopping=None, use_missing=False, zero_as_missing=False,
                  complete_secure=False, tree_num_per_party=1, guest_depth=1, host_depth=1, work_mode='mix', metrics=None,
-                 sparse_optmization=False, subsample_random_seed=None, binning_error=consts.DEFAULT_RELATIVE_ERROR):
+                 sparse_optimization=False, subsample_random_seed=None, binning_error=consts.DEFAULT_RELATIVE_ERROR):
 
         """
         work_mode：
@@ -492,7 +492,7 @@ def __init__(self, tree_param: DecisionTreeParam = DecisionTreeParam(), task_typ
                                                          predict_param, cv_param, validation_freqs, early_stopping,
                                                          use_missing, zero_as_missing, complete_secure, metrics=metrics,
                                                          subsample_random_seed=subsample_random_seed,
-                                                         sparse_optmization=sparse_optmization,
+                                                         sparse_optimization=sparse_optimization,
                                                          binning_error=binning_error)
 
         self.tree_num_per_party = tree_num_per_party