fate2.x HeteroSecureBoost 无法设置 early_stopping_rounds 参数

[wangshoujunnew]

#源码部分

class HeteroSecureBoost(Component):
    yaml_define_path = "./component_define/fate/hetero_secureboost.yaml"
def __init__(
    self,
    _name: str,
    runtime_parties: dict = None,
    train_data: ArtifactType = PlaceHolder(),
    validate_data: ArtifactType = PlaceHolder(),
    num_trees: int = 20,
    learning_rate: float = 0.3,
    max_depth: int = 3,
    max_bin: int = 32,
    objective: str = "binary:bce",
    num_class: int = 2,
    goss: bool = False,
    goss_start_iter: int = 0,
    top_rate: float = 0.2,
    other_rate: float = 0.1,
    l1: float = 0,
    l2: float = 0.1,
    min_impurity_split: float = 1e-2,
    min_sample_split: int = 2,
    min_leaf_node: int = 1,
    min_child_weight: float = 1,
    gh_pack: bool = True,
    split_info_pack: bool = True,
    hist_sub: bool = True,
    he_param: dict = PlaceHolder(),
    cv_param: dict = PlaceHolder(),
    train_output_data: ArtifactType = PlaceHolder(),
    output_model: ArtifactType = PlaceHolder(),
    warm_start_model: ArtifactType = PlaceHolder(),
    test_data: ArtifactType = PlaceHolder(),
    input_model: ArtifactType = PlaceHolder(),
    cv_data: ArtifactType = PlaceHolder()
)

================

hetero_sbt_0 = HeteroSecureBoost('sbt_0', num_trees=10, max_bin=32, max_depth=5, goss=True, top_rate=0.2,
he_param={'kind': 'paillier', 'key_length': 1024},
train_data=psi_0.outputs['output_data'],
validate_data=psi_1.outputs['output_data'],
early_stop=2
)

Traceback (most recent call last):
File "/root/test_train.py", line 100, in
main(config=args.config, namespace=args.namespace)
File "/root/test_train.py", line 53, in main
hetero_sbt_0 = HeteroSecureBoost('sbt_0', num_trees=10, max_bin=32, max_depth=5, goss=True, top_rate=0.2,
TypeError: HeteroSecureBoost.init() got an unexpected keyword argument 'early_stop'