# -*- 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)