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

Classes:

LevelWiseCache()

class secretflow.ml.boost.sgb_v.factory.components.cache.LevelWiseCache[source]#

Bases: Component

Methods:

__init__()

show_params()

set_params(_)

get_params(_)

set_devices(devices)

reset_level_caches()

collect_level_node_GH(worker, bucket_sums, ...)

get_level_nodes_GH(worker)

update_level_cache(is_last_level, ...)

__init__()[source]#
show_params()[source]#
set_params(_: dict)[source]#
get_params(_: dict)[source]#
set_devices(devices: Devices)[source]#
reset_level_caches()[source]#
collect_level_node_GH(worker, bucket_sums, is_lefts)[source]#
get_level_nodes_GH(worker) → List[source]#
update_level_cache(is_last_level, gain_is_cost_effective)[source]#

secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache#

Classes:

LevelWiseCache()

class secretflow.ml.boost.sgb_v.factory.components.cache.level_wise_cache.LevelWiseCache[source]#

Bases: Component

Methods:

__init__()

show_params()

set_params(_)

get_params(_)

set_devices(devices)

reset_level_caches()

collect_level_node_GH(worker, bucket_sums, ...)

get_level_nodes_GH(worker)

update_level_cache(is_last_level, ...)

__init__()[source]#
show_params()[source]#
set_params(_: dict)[source]#
get_params(_: dict)[source]#
set_devices(devices: Devices)[source]#
reset_level_caches()[source]#
collect_level_node_GH(worker, bucket_sums, is_lefts)[source]#
get_level_nodes_GH(worker) → List[source]#
update_level_cache(is_last_level, gain_is_cost_effective)[source]#