secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator#
Classes:
- class secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.BucketSumCalculator[源代码]#
基类:
Composite
Methods:
__init__
()set_params
(_)get_params
(_)set_devices
(devices)calculate_bucket_sum_level_wise
(shuffler, ...)update_level_cache
(is_last_level, ...)- calculate_bucket_sum_level_wise(shuffler: Shuffler, encrypted_gh_dict: Dict[PYU, HEUObject], children_split_node_selects: List[PYUObject], is_lefts: List[bool], order_map_sub: FedNdarray, bucket_num: int, bucket_lists: List[PYUObject], gradient_encryptor: GradientEncryptor, node_num: int) Tuple[PYUObject, PYUObject] [源代码]#
secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.bucket_sum_calculator#
Classes:
|
|
- class secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.bucket_sum_calculator.BucketSumCalculatorComponents(level_wise_cache: secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache = <secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache object at 0x7f5e4f0a36d0>)[源代码]#
基类:
object
Attributes:
Methods:
__init__
([level_wise_cache])- level_wise_cache: LevelWiseCache = <secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache object>#
- __init__(level_wise_cache: ~secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache = <secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache object>) None #
- class secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.bucket_sum_calculator.BucketSumCalculator[源代码]#
基类:
Composite
Methods:
__init__
()set_params
(_)get_params
(_)set_devices
(devices)calculate_bucket_sum_level_wise
(shuffler, ...)update_level_cache
(is_last_level, ...)- calculate_bucket_sum_level_wise(shuffler: Shuffler, encrypted_gh_dict: Dict[PYU, HEUObject], children_split_node_selects: List[PYUObject], is_lefts: List[bool], order_map_sub: FedNdarray, bucket_num: int, bucket_lists: List[PYUObject], gradient_encryptor: GradientEncryptor, node_num: int) Tuple[PYUObject, PYUObject] [源代码]#