hetero_nn_model_meta_pb2.py 3.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: hetero-nn-model-meta.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\x1ahetero-nn-model-meta.proto\x12&com.webank.ai.fate.core.mlmodel.buffer\"3\n\x0eOptimizerParam\x12\x11\n\toptimizer\x18\x01 \x01(\t\x12\x0e\n\x06kwargs\x18\x02 \x01(\t\"!\n\x0cPredictParam\x12\x11\n\tthreshold\x18\x01 \x01(\x01\"\x89\x02\n\x11HeteroNNModelMeta\x12\x13\n\x0b\x63onfig_type\x18\x01 \x01(\t\x12\x18\n\x10\x62ottom_nn_define\x18\x02 \x03(\t\x12 \n\x18interactive_layer_define\x18\x03 \x01(\t\x12\x15\n\rtop_nn_define\x18\x04 \x03(\t\x12\x0f\n\x07metrics\x18\x05 \x03(\t\x12O\n\x0foptimizer_param\x18\x06 \x01(\x0b\x32\x36.com.webank.ai.fate.core.mlmodel.buffer.OptimizerParam\x12\x0c\n\x04loss\x18\x07 \x01(\t\x12\x1c\n\x14interactive_layer_lr\x18\x08 \x01(\x01\"\xcf\x01\n\x0cHeteroNNMeta\x12W\n\x14hetero_nn_model_meta\x18\x01 \x01(\x0b\x32\x39.com.webank.ai.fate.core.mlmodel.buffer.HeteroNNModelMeta\x12\x11\n\ttask_type\x18\x02 \x01(\t\x12\x12\n\nbatch_size\x18\x03 \x01(\x05\x12\x0e\n\x06\x65pochs\x18\x04 \x01(\x05\x12\x12\n\nearly_stop\x18\x05 \x01(\t\x12\x0b\n\x03tol\x18\x06 \x01(\x01\x12\x0e\n\x06module\x18\x07 \x01(\tB\x18\x42\x16HeteroNNModelMetaProtob\x06proto3')
  13. _OPTIMIZERPARAM = DESCRIPTOR.message_types_by_name['OptimizerParam']
  14. _PREDICTPARAM = DESCRIPTOR.message_types_by_name['PredictParam']
  15. _HETERONNMODELMETA = DESCRIPTOR.message_types_by_name['HeteroNNModelMeta']
  16. _HETERONNMETA = DESCRIPTOR.message_types_by_name['HeteroNNMeta']
  17. OptimizerParam = _reflection.GeneratedProtocolMessageType('OptimizerParam', (_message.Message,), {
  18. 'DESCRIPTOR' : _OPTIMIZERPARAM,
  19. '__module__' : 'hetero_nn_model_meta_pb2'
  20. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.OptimizerParam)
  21. })
  22. _sym_db.RegisterMessage(OptimizerParam)
  23. PredictParam = _reflection.GeneratedProtocolMessageType('PredictParam', (_message.Message,), {
  24. 'DESCRIPTOR' : _PREDICTPARAM,
  25. '__module__' : 'hetero_nn_model_meta_pb2'
  26. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.PredictParam)
  27. })
  28. _sym_db.RegisterMessage(PredictParam)
  29. HeteroNNModelMeta = _reflection.GeneratedProtocolMessageType('HeteroNNModelMeta', (_message.Message,), {
  30. 'DESCRIPTOR' : _HETERONNMODELMETA,
  31. '__module__' : 'hetero_nn_model_meta_pb2'
  32. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.HeteroNNModelMeta)
  33. })
  34. _sym_db.RegisterMessage(HeteroNNModelMeta)
  35. HeteroNNMeta = _reflection.GeneratedProtocolMessageType('HeteroNNMeta', (_message.Message,), {
  36. 'DESCRIPTOR' : _HETERONNMETA,
  37. '__module__' : 'hetero_nn_model_meta_pb2'
  38. # @@protoc_insertion_point(class_scope:com.webank.ai.fate.core.mlmodel.buffer.HeteroNNMeta)
  39. })
  40. _sym_db.RegisterMessage(HeteroNNMeta)
  41. if _descriptor._USE_C_DESCRIPTORS == False:
  42. DESCRIPTOR._options = None
  43. DESCRIPTOR._serialized_options = b'B\026HeteroNNModelMetaProto'
  44. _OPTIMIZERPARAM._serialized_start=70
  45. _OPTIMIZERPARAM._serialized_end=121
  46. _PREDICTPARAM._serialized_start=123
  47. _PREDICTPARAM._serialized_end=156
  48. _HETERONNMODELMETA._serialized_start=159
  49. _HETERONNMODELMETA._serialized_end=424
  50. _HETERONNMETA._serialized_start=427
  51. _HETERONNMETA._serialized_end=634
  52. # @@protoc_insertion_point(module_scope)