secretflow.ml.boost.sgb_v.factory.components.model_builder#
Classes:
Functions related to build models including making predictions and add trees |
- class secretflow.ml.boost.sgb_v.factory.components.model_builder.ModelBuilder[源代码]#
基类:
Component
Functions related to build models including making predictions and add trees
Methods:
__init__
()set_params
(params)get_params
(params)set_devices
(devices)init_pred
(sample_num)insert_tree
(tree)finish
()- insert_tree(tree: DistributedTree)[源代码]#
secretflow.ml.boost.sgb_v.factory.components.model_builder.model_builder#
Classes:
|
'objective': Specify the learning objective. |
Functions related to build models including making predictions and add trees |
- class secretflow.ml.boost.sgb_v.factory.components.model_builder.model_builder.ModelBuilderParams(base_score: float = 0, objective: RegType = RegType.Logistic)[源代码]#
基类:
object
- ‘objective’: Specify the learning objective.
default: ‘logistic’ range: [‘linear’, ‘logistic’]
- ‘base_score’: The initial prediction score of all instances, global bias.
default: 0
Attributes:
Methods:
__init__
([base_score, objective])- base_score: float = 0#
- class secretflow.ml.boost.sgb_v.factory.components.model_builder.model_builder.ModelBuilder[源代码]#
基类:
Component
Functions related to build models including making predictions and add trees
Methods:
__init__
()set_params
(params)get_params
(params)set_devices
(devices)init_pred
(sample_num)insert_tree
(tree)finish
()- insert_tree(tree: DistributedTree)[源代码]#