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[源代码]#
基类:
ComponentFunctions 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[源代码]#
基类:
ComponentFunctions 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)[源代码]#