secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator#

Classes:

BucketSumCalculator()

class secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.BucketSumCalculator[源代码]#

基类:Composite

Methods:

__init__()

show_params()

set_params(_)

get_params(_)

set_devices(devices)

calculate_bucket_sum_level_wise(shuffler, ...)

update_level_cache(is_last_level, ...)

__init__()[源代码]#
show_params()[源代码]#
set_params(_: dict)[源代码]#
get_params(_: dict)[源代码]#
set_devices(devices: Devices)[源代码]#
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][源代码]#
update_level_cache(is_last_level, gain_is_cost_effective)[源代码]#

secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.bucket_sum_calculator#

Classes:

BucketSumCalculatorComponents(level_wise_cache)

BucketSumCalculator()

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:

level_wise_cache

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__()

show_params()

set_params(_)

get_params(_)

set_devices(devices)

calculate_bucket_sum_level_wise(shuffler, ...)

update_level_cache(is_last_level, ...)

__init__()[源代码]#
show_params()[源代码]#
set_params(_: dict)[源代码]#
get_params(_: dict)[源代码]#
set_devices(devices: Devices)[源代码]#
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][源代码]#
update_level_cache(is_last_level, gain_is_cost_effective)[源代码]#