123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140 |
- # -*- coding: utf-8 -*-
- # Generated by the protocol buffer compiler. DO NOT EDIT!
- # source: boosting-tree-model-param.proto
- """Generated protocol buffer code."""
- from google.protobuf import descriptor as _descriptor
- from google.protobuf import descriptor_pool as _descriptor_pool
- from google.protobuf import message as _message
- from google.protobuf import reflection as _reflection
- from google.protobuf import symbol_database as _symbol_database
- # @@protoc_insertion_point(imports)
- _sym_db = _symbol_database.Default()
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1f\x62oosting-tree-model-param.proto\x12&com.webank.ai.fate.core.mlmodel.buffer\"\xb7\x01\n\tNodeParam\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x10\n\x08sitename\x18\x02 \x01(\t\x12\x0b\n\x03\x66id\x18\x03 \x01(\x05\x12\x0b\n\x03\x62id\x18\x04 \x01(\x01\x12\x0e\n\x06weight\x18\x05 \x01(\x01\x12\x0f\n\x07is_leaf\x18\x06 \x01(\x08\x12\x13\n\x0bleft_nodeid\x18\x07 \x01(\x05\x12\x14\n\x0cright_nodeid\x18\x08 \x01(\x05\x12\x13\n\x0bmissing_dir\x18\t \x01(\x05\x12\x11\n\tmo_weight\x18\n \x03(\x01\"\xc1\x04\n\x16\x44\x65\x63isionTreeModelParam\x12@\n\x05tree_\x18\x01 \x03(\x0b\x32\x31.com.webank.ai.fate.core.mlmodel.buffer.NodeParam\x12i\n\x0esplit_maskdict\x18\x02 \x03(\x0b\x32Q.com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.SplitMaskdictEntry\x12t\n\x14missing_dir_maskdict\x18\x03 \x03(\x0b\x32V.com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.MissingDirMaskdictEntry\x12\x61\n\nleaf_count\x18\x04 \x03(\x0b\x32M.com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.LeafCountEntry\x1a\x34\n\x12SplitMaskdictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x39\n\x17MissingDirMaskdictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\x1a\x30\n\x0eLeafCountEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\x7f\n\x15\x46\x65\x61tureImportanceInfo\x12\x10\n\x08sitename\x18\x01 \x01(\t\x12\x0b\n\x03\x66id\x18\x02 \x01(\x05\x12\x12\n\nimportance\x18\x03 \x01(\x01\x12\x10\n\x08\x66ullname\x18\x04 \x01(\t\x12\x13\n\x0bimportance2\x18\x05 \x01(\x01\x12\x0c\n\x04main\x18\x06 \x01(\t\"\xe4\x05\n\x16\x42oostingTreeModelParam\x12\x10\n\x08tree_num\x18\x01 \x01(\x05\x12N\n\x06trees_\x18\x02 \x03(\x0b\x32>.com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam\x12\x12\n\ninit_score\x18\x03 \x03(\x01\x12\x0e\n\x06losses\x18\x04 \x03(\x01\x12\x10\n\x08tree_dim\x18\x05 \x01(\x05\x12\x13\n\x0bnum_classes\x18\x06 \x01(\x05\x12\x10\n\x08\x63lasses_\x18\x07 \x03(\t\x12Z\n\x13\x66\x65\x61ture_importances\x18\x08 \x03(\x0b\x32=.com.webank.ai.fate.core.mlmodel.buffer.FeatureImportanceInfo\x12{\n\x18\x66\x65\x61ture_name_fid_mapping\x18\t \x03(\x0b\x32Y.com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.FeatureNameFidMappingEntry\x12\x16\n\x0e\x62\x65st_iteration\x18\n \x01(\x05\x12\x11\n\ttree_plan\x18\x0b \x03(\t\x12\x12\n\nmodel_name\x18\x0c \x01(\t\x12x\n\x16\x61nonymous_name_mapping\x18\r \x03(\x0b\x32X.com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.AnonymousNameMappingEntry\x1a<\n\x1a\x46\x65\x61tureNameFidMappingEntry\x12\x0b\n\x03key\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a;\n\x19\x41nonymousNameMappingEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"z\n\x10TransformerParam\x12R\n\ntree_param\x18\x01 \x01(\x0b\x32>.com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam\x12\x12\n\nmodel_name\x18\x02 \x01(\tB\x1a\x42\x18\x42oostTreeModelParamProtob\x06proto3')
- _NODEPARAM = DESCRIPTOR.message_types_by_name['NodeParam']
- _DECISIONTREEMODELPARAM = DESCRIPTOR.message_types_by_name['DecisionTreeModelParam']
- _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['SplitMaskdictEntry']
- _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['MissingDirMaskdictEntry']
- _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['LeafCountEntry']
- _FEATUREIMPORTANCEINFO = DESCRIPTOR.message_types_by_name['FeatureImportanceInfo']
- _BOOSTINGTREEMODELPARAM = DESCRIPTOR.message_types_by_name['BoostingTreeModelParam']
- _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY = _BOOSTINGTREEMODELPARAM.nested_types_by_name['FeatureNameFidMappingEntry']
- _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY = _BOOSTINGTREEMODELPARAM.nested_types_by_name['AnonymousNameMappingEntry']
- _TRANSFORMERPARAM = DESCRIPTOR.message_types_by_name['TransformerParam']
- NodeParam = _reflection.GeneratedProtocolMessageType('NodeParam', (_message.Message,), {
- 'DESCRIPTOR' : _NODEPARAM,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.NodeParam)
- })
- _sym_db.RegisterMessage(NodeParam)
- DecisionTreeModelParam = _reflection.GeneratedProtocolMessageType('DecisionTreeModelParam', (_message.Message,), {
- 'SplitMaskdictEntry' : _reflection.GeneratedProtocolMessageType('SplitMaskdictEntry', (_message.Message,), {
- 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.SplitMaskdictEntry)
- })
- ,
- 'MissingDirMaskdictEntry' : _reflection.GeneratedProtocolMessageType('MissingDirMaskdictEntry', (_message.Message,), {
- 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.MissingDirMaskdictEntry)
- })
- ,
- 'LeafCountEntry' : _reflection.GeneratedProtocolMessageType('LeafCountEntry', (_message.Message,), {
- 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.LeafCountEntry)
- })
- ,
- 'DESCRIPTOR' : _DECISIONTREEMODELPARAM,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam)
- })
- _sym_db.RegisterMessage(DecisionTreeModelParam)
- _sym_db.RegisterMessage(DecisionTreeModelParam.SplitMaskdictEntry)
- _sym_db.RegisterMessage(DecisionTreeModelParam.MissingDirMaskdictEntry)
- _sym_db.RegisterMessage(DecisionTreeModelParam.LeafCountEntry)
- FeatureImportanceInfo = _reflection.GeneratedProtocolMessageType('FeatureImportanceInfo', (_message.Message,), {
- 'DESCRIPTOR' : _FEATUREIMPORTANCEINFO,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.FeatureImportanceInfo)
- })
- _sym_db.RegisterMessage(FeatureImportanceInfo)
- BoostingTreeModelParam = _reflection.GeneratedProtocolMessageType('BoostingTreeModelParam', (_message.Message,), {
- 'FeatureNameFidMappingEntry' : _reflection.GeneratedProtocolMessageType('FeatureNameFidMappingEntry', (_message.Message,), {
- 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.FeatureNameFidMappingEntry)
- })
- ,
- 'AnonymousNameMappingEntry' : _reflection.GeneratedProtocolMessageType('AnonymousNameMappingEntry', (_message.Message,), {
- 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.AnonymousNameMappingEntry)
- })
- ,
- 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam)
- })
- _sym_db.RegisterMessage(BoostingTreeModelParam)
- _sym_db.RegisterMessage(BoostingTreeModelParam.FeatureNameFidMappingEntry)
- _sym_db.RegisterMessage(BoostingTreeModelParam.AnonymousNameMappingEntry)
- TransformerParam = _reflection.GeneratedProtocolMessageType('TransformerParam', (_message.Message,), {
- 'DESCRIPTOR' : _TRANSFORMERPARAM,
- '__module__' : 'boosting_tree_model_param_pb2'
- # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.TransformerParam)
- })
- _sym_db.RegisterMessage(TransformerParam)
- if _descriptor._USE_C_DESCRIPTORS == False:
- DESCRIPTOR._options = None
- DESCRIPTOR._serialized_options = b'B\030BoostTreeModelParamProto'
- _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._options = None
- _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_options = b'8\001'
- _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._options = None
- _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_options = b'8\001'
- _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._options = None
- _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_options = b'8\001'
- _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._options = None
- _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_options = b'8\001'
- _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._options = None
- _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_options = b'8\001'
- _NODEPARAM._serialized_start=76
- _NODEPARAM._serialized_end=259
- _DECISIONTREEMODELPARAM._serialized_start=262
- _DECISIONTREEMODELPARAM._serialized_end=839
- _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_start=678
- _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_end=730
- _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_start=732
- _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_end=789
- _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_start=791
- _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_end=839
- _FEATUREIMPORTANCEINFO._serialized_start=841
- _FEATUREIMPORTANCEINFO._serialized_end=968
- _BOOSTINGTREEMODELPARAM._serialized_start=971
- _BOOSTINGTREEMODELPARAM._serialized_end=1711
- _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_start=1590
- _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_end=1650
- _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_start=1652
- _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_end=1711
- _TRANSFORMERPARAM._serialized_start=1713
- _TRANSFORMERPARAM._serialized_end=1835
- # @@protoc_insertion_point(module_scope)
|