boosting_tree_model_param_pb2.py 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: boosting-tree-model-param.proto
  4. """Generated protocol buffer code."""
  5. from google.protobuf import descriptor as _descriptor
  6. from google.protobuf import descriptor_pool as _descriptor_pool
  7. from google.protobuf import message as _message
  8. from google.protobuf import reflection as _reflection
  9. from google.protobuf import symbol_database as _symbol_database
  10. # @@protoc_insertion_point(imports)
  11. _sym_db = _symbol_database.Default()
  12. 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')
  13. _NODEPARAM = DESCRIPTOR.message_types_by_name['NodeParam']
  14. _DECISIONTREEMODELPARAM = DESCRIPTOR.message_types_by_name['DecisionTreeModelParam']
  15. _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['SplitMaskdictEntry']
  16. _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['MissingDirMaskdictEntry']
  17. _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY = _DECISIONTREEMODELPARAM.nested_types_by_name['LeafCountEntry']
  18. _FEATUREIMPORTANCEINFO = DESCRIPTOR.message_types_by_name['FeatureImportanceInfo']
  19. _BOOSTINGTREEMODELPARAM = DESCRIPTOR.message_types_by_name['BoostingTreeModelParam']
  20. _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY = _BOOSTINGTREEMODELPARAM.nested_types_by_name['FeatureNameFidMappingEntry']
  21. _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY = _BOOSTINGTREEMODELPARAM.nested_types_by_name['AnonymousNameMappingEntry']
  22. _TRANSFORMERPARAM = DESCRIPTOR.message_types_by_name['TransformerParam']
  23. NodeParam = _reflection.GeneratedProtocolMessageType('NodeParam', (_message.Message,), {
  24. 'DESCRIPTOR' : _NODEPARAM,
  25. '__module__' : 'boosting_tree_model_param_pb2'
  26. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.NodeParam)
  27. })
  28. _sym_db.RegisterMessage(NodeParam)
  29. DecisionTreeModelParam = _reflection.GeneratedProtocolMessageType('DecisionTreeModelParam', (_message.Message,), {
  30. 'SplitMaskdictEntry' : _reflection.GeneratedProtocolMessageType('SplitMaskdictEntry', (_message.Message,), {
  31. 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY,
  32. '__module__' : 'boosting_tree_model_param_pb2'
  33. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.SplitMaskdictEntry)
  34. })
  35. ,
  36. 'MissingDirMaskdictEntry' : _reflection.GeneratedProtocolMessageType('MissingDirMaskdictEntry', (_message.Message,), {
  37. 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY,
  38. '__module__' : 'boosting_tree_model_param_pb2'
  39. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.MissingDirMaskdictEntry)
  40. })
  41. ,
  42. 'LeafCountEntry' : _reflection.GeneratedProtocolMessageType('LeafCountEntry', (_message.Message,), {
  43. 'DESCRIPTOR' : _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY,
  44. '__module__' : 'boosting_tree_model_param_pb2'
  45. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam.LeafCountEntry)
  46. })
  47. ,
  48. 'DESCRIPTOR' : _DECISIONTREEMODELPARAM,
  49. '__module__' : 'boosting_tree_model_param_pb2'
  50. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.DecisionTreeModelParam)
  51. })
  52. _sym_db.RegisterMessage(DecisionTreeModelParam)
  53. _sym_db.RegisterMessage(DecisionTreeModelParam.SplitMaskdictEntry)
  54. _sym_db.RegisterMessage(DecisionTreeModelParam.MissingDirMaskdictEntry)
  55. _sym_db.RegisterMessage(DecisionTreeModelParam.LeafCountEntry)
  56. FeatureImportanceInfo = _reflection.GeneratedProtocolMessageType('FeatureImportanceInfo', (_message.Message,), {
  57. 'DESCRIPTOR' : _FEATUREIMPORTANCEINFO,
  58. '__module__' : 'boosting_tree_model_param_pb2'
  59. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.FeatureImportanceInfo)
  60. })
  61. _sym_db.RegisterMessage(FeatureImportanceInfo)
  62. BoostingTreeModelParam = _reflection.GeneratedProtocolMessageType('BoostingTreeModelParam', (_message.Message,), {
  63. 'FeatureNameFidMappingEntry' : _reflection.GeneratedProtocolMessageType('FeatureNameFidMappingEntry', (_message.Message,), {
  64. 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY,
  65. '__module__' : 'boosting_tree_model_param_pb2'
  66. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.FeatureNameFidMappingEntry)
  67. })
  68. ,
  69. 'AnonymousNameMappingEntry' : _reflection.GeneratedProtocolMessageType('AnonymousNameMappingEntry', (_message.Message,), {
  70. 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY,
  71. '__module__' : 'boosting_tree_model_param_pb2'
  72. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam.AnonymousNameMappingEntry)
  73. })
  74. ,
  75. 'DESCRIPTOR' : _BOOSTINGTREEMODELPARAM,
  76. '__module__' : 'boosting_tree_model_param_pb2'
  77. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.BoostingTreeModelParam)
  78. })
  79. _sym_db.RegisterMessage(BoostingTreeModelParam)
  80. _sym_db.RegisterMessage(BoostingTreeModelParam.FeatureNameFidMappingEntry)
  81. _sym_db.RegisterMessage(BoostingTreeModelParam.AnonymousNameMappingEntry)
  82. TransformerParam = _reflection.GeneratedProtocolMessageType('TransformerParam', (_message.Message,), {
  83. 'DESCRIPTOR' : _TRANSFORMERPARAM,
  84. '__module__' : 'boosting_tree_model_param_pb2'
  85. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.TransformerParam)
  86. })
  87. _sym_db.RegisterMessage(TransformerParam)
  88. if _descriptor._USE_C_DESCRIPTORS == False:
  89. DESCRIPTOR._options = None
  90. DESCRIPTOR._serialized_options = b'B\030BoostTreeModelParamProto'
  91. _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._options = None
  92. _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_options = b'8\001'
  93. _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._options = None
  94. _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_options = b'8\001'
  95. _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._options = None
  96. _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_options = b'8\001'
  97. _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._options = None
  98. _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_options = b'8\001'
  99. _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._options = None
  100. _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_options = b'8\001'
  101. _NODEPARAM._serialized_start=76
  102. _NODEPARAM._serialized_end=259
  103. _DECISIONTREEMODELPARAM._serialized_start=262
  104. _DECISIONTREEMODELPARAM._serialized_end=839
  105. _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_start=678
  106. _DECISIONTREEMODELPARAM_SPLITMASKDICTENTRY._serialized_end=730
  107. _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_start=732
  108. _DECISIONTREEMODELPARAM_MISSINGDIRMASKDICTENTRY._serialized_end=789
  109. _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_start=791
  110. _DECISIONTREEMODELPARAM_LEAFCOUNTENTRY._serialized_end=839
  111. _FEATUREIMPORTANCEINFO._serialized_start=841
  112. _FEATUREIMPORTANCEINFO._serialized_end=968
  113. _BOOSTINGTREEMODELPARAM._serialized_start=971
  114. _BOOSTINGTREEMODELPARAM._serialized_end=1711
  115. _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_start=1590
  116. _BOOSTINGTREEMODELPARAM_FEATURENAMEFIDMAPPINGENTRY._serialized_end=1650
  117. _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_start=1652
  118. _BOOSTINGTREEMODELPARAM_ANONYMOUSNAMEMAPPINGENTRY._serialized_end=1711
  119. _TRANSFORMERPARAM._serialized_start=1713
  120. _TRANSFORMERPARAM._serialized_end=1835
  121. # @@protoc_insertion_point(module_scope)