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

Classes:

BucketSumCalculator()

class secretflow.ml.boost.sgb_v.factory.components.bucket_sum_calculator.BucketSumCalculator[source]#

Bases: 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__()[source]#

show_params()[source]#

set_params(_: dict)[source]#

get_params(_: dict)[source]#

set_devices(devices: Devices)[source]#

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][source]#

update_level_cache(is_last_level, gain_is_cost_effective)[source]#

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 0x7f47491802b0>)[source]#

Bases: 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[source]#

Bases: 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__()[source]#

show_params()[source]#

set_params(_: dict)[source]#

get_params(_: dict)[source]#

set_devices(devices: Devices)[source]#

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][source]#

update_level_cache(is_last_level, gain_is_cost_effective)[source]#