easy-cs-rec-custommodel 0.8.6__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of easy-cs-rec-custommodel might be problematic. Click here for more details.

Files changed (336) hide show
  1. easy_cs_rec_custommodel-0.8.6.dist-info/LICENSE +203 -0
  2. easy_cs_rec_custommodel-0.8.6.dist-info/METADATA +48 -0
  3. easy_cs_rec_custommodel-0.8.6.dist-info/RECORD +336 -0
  4. easy_cs_rec_custommodel-0.8.6.dist-info/WHEEL +6 -0
  5. easy_cs_rec_custommodel-0.8.6.dist-info/top_level.txt +2 -0
  6. easy_rec/__init__.py +114 -0
  7. easy_rec/python/__init__.py +0 -0
  8. easy_rec/python/builders/__init__.py +0 -0
  9. easy_rec/python/builders/hyperparams_builder.py +78 -0
  10. easy_rec/python/builders/loss_builder.py +333 -0
  11. easy_rec/python/builders/optimizer_builder.py +211 -0
  12. easy_rec/python/builders/strategy_builder.py +44 -0
  13. easy_rec/python/compat/__init__.py +0 -0
  14. easy_rec/python/compat/adam_s.py +245 -0
  15. easy_rec/python/compat/array_ops.py +229 -0
  16. easy_rec/python/compat/dynamic_variable.py +542 -0
  17. easy_rec/python/compat/early_stopping.py +653 -0
  18. easy_rec/python/compat/embedding_ops.py +162 -0
  19. easy_rec/python/compat/embedding_parallel_saver.py +316 -0
  20. easy_rec/python/compat/estimator_train.py +116 -0
  21. easy_rec/python/compat/exporter.py +473 -0
  22. easy_rec/python/compat/feature_column/__init__.py +0 -0
  23. easy_rec/python/compat/feature_column/feature_column.py +3675 -0
  24. easy_rec/python/compat/feature_column/feature_column_v2.py +5233 -0
  25. easy_rec/python/compat/feature_column/sequence_feature_column.py +648 -0
  26. easy_rec/python/compat/feature_column/utils.py +154 -0
  27. easy_rec/python/compat/layers.py +329 -0
  28. easy_rec/python/compat/ops.py +14 -0
  29. easy_rec/python/compat/optimizers.py +619 -0
  30. easy_rec/python/compat/queues.py +311 -0
  31. easy_rec/python/compat/regularizers.py +208 -0
  32. easy_rec/python/compat/sok_optimizer.py +440 -0
  33. easy_rec/python/compat/sync_replicas_optimizer.py +528 -0
  34. easy_rec/python/compat/weight_decay_optimizers.py +475 -0
  35. easy_rec/python/core/__init__.py +0 -0
  36. easy_rec/python/core/easyrec_metrics/__init__.py +24 -0
  37. easy_rec/python/core/easyrec_metrics/distribute_metrics_impl_pai.py +3702 -0
  38. easy_rec/python/core/easyrec_metrics/distribute_metrics_impl_tf.py +3768 -0
  39. easy_rec/python/core/learning_schedules.py +228 -0
  40. easy_rec/python/core/metrics.py +402 -0
  41. easy_rec/python/core/sampler.py +844 -0
  42. easy_rec/python/eval.py +102 -0
  43. easy_rec/python/export.py +150 -0
  44. easy_rec/python/feature_column/__init__.py +0 -0
  45. easy_rec/python/feature_column/feature_column.py +664 -0
  46. easy_rec/python/feature_column/feature_group.py +89 -0
  47. easy_rec/python/hpo/__init__.py +0 -0
  48. easy_rec/python/hpo/emr_hpo.py +140 -0
  49. easy_rec/python/hpo/generate_hpo_sql.py +71 -0
  50. easy_rec/python/hpo/pai_hpo.py +297 -0
  51. easy_rec/python/inference/__init__.py +0 -0
  52. easy_rec/python/inference/csv_predictor.py +189 -0
  53. easy_rec/python/inference/hive_parquet_predictor.py +200 -0
  54. easy_rec/python/inference/hive_predictor.py +166 -0
  55. easy_rec/python/inference/odps_predictor.py +70 -0
  56. easy_rec/python/inference/parquet_predictor.py +147 -0
  57. easy_rec/python/inference/parquet_predictor_v2.py +147 -0
  58. easy_rec/python/inference/predictor.py +621 -0
  59. easy_rec/python/inference/processor/__init__.py +0 -0
  60. easy_rec/python/inference/processor/test.py +170 -0
  61. easy_rec/python/inference/vector_retrieve.py +124 -0
  62. easy_rec/python/input/__init__.py +0 -0
  63. easy_rec/python/input/batch_tfrecord_input.py +117 -0
  64. easy_rec/python/input/criteo_binary_reader.py +259 -0
  65. easy_rec/python/input/criteo_input.py +107 -0
  66. easy_rec/python/input/csv_input.py +175 -0
  67. easy_rec/python/input/csv_input_ex.py +72 -0
  68. easy_rec/python/input/csv_input_v2.py +68 -0
  69. easy_rec/python/input/datahub_input.py +320 -0
  70. easy_rec/python/input/dummy_input.py +58 -0
  71. easy_rec/python/input/hive_input.py +123 -0
  72. easy_rec/python/input/hive_parquet_input.py +140 -0
  73. easy_rec/python/input/hive_rtp_input.py +174 -0
  74. easy_rec/python/input/input.py +1064 -0
  75. easy_rec/python/input/kafka_dataset.py +144 -0
  76. easy_rec/python/input/kafka_input.py +235 -0
  77. easy_rec/python/input/load_parquet.py +317 -0
  78. easy_rec/python/input/odps_input.py +101 -0
  79. easy_rec/python/input/odps_input_v2.py +110 -0
  80. easy_rec/python/input/odps_input_v3.py +132 -0
  81. easy_rec/python/input/odps_rtp_input.py +187 -0
  82. easy_rec/python/input/odps_rtp_input_v2.py +104 -0
  83. easy_rec/python/input/parquet_input.py +397 -0
  84. easy_rec/python/input/parquet_input_v2.py +180 -0
  85. easy_rec/python/input/parquet_input_v3.py +203 -0
  86. easy_rec/python/input/rtp_input.py +225 -0
  87. easy_rec/python/input/rtp_input_v2.py +145 -0
  88. easy_rec/python/input/tfrecord_input.py +100 -0
  89. easy_rec/python/layers/__init__.py +0 -0
  90. easy_rec/python/layers/backbone.py +571 -0
  91. easy_rec/python/layers/capsule_layer.py +176 -0
  92. easy_rec/python/layers/cmbf.py +390 -0
  93. easy_rec/python/layers/common_layers.py +192 -0
  94. easy_rec/python/layers/dnn.py +87 -0
  95. easy_rec/python/layers/embed_input_layer.py +25 -0
  96. easy_rec/python/layers/fm.py +26 -0
  97. easy_rec/python/layers/input_layer.py +396 -0
  98. easy_rec/python/layers/keras/__init__.py +34 -0
  99. easy_rec/python/layers/keras/activation.py +114 -0
  100. easy_rec/python/layers/keras/attention.py +267 -0
  101. easy_rec/python/layers/keras/auxiliary_loss.py +47 -0
  102. easy_rec/python/layers/keras/blocks.py +262 -0
  103. easy_rec/python/layers/keras/bst.py +119 -0
  104. easy_rec/python/layers/keras/custom_ops.py +250 -0
  105. easy_rec/python/layers/keras/data_augment.py +133 -0
  106. easy_rec/python/layers/keras/din.py +67 -0
  107. easy_rec/python/layers/keras/einsum_dense.py +598 -0
  108. easy_rec/python/layers/keras/embedding.py +81 -0
  109. easy_rec/python/layers/keras/fibinet.py +251 -0
  110. easy_rec/python/layers/keras/interaction.py +416 -0
  111. easy_rec/python/layers/keras/layer_norm.py +364 -0
  112. easy_rec/python/layers/keras/mask_net.py +166 -0
  113. easy_rec/python/layers/keras/multi_head_attention.py +717 -0
  114. easy_rec/python/layers/keras/multi_task.py +125 -0
  115. easy_rec/python/layers/keras/numerical_embedding.py +376 -0
  116. easy_rec/python/layers/keras/ppnet.py +194 -0
  117. easy_rec/python/layers/keras/transformer.py +192 -0
  118. easy_rec/python/layers/layer_norm.py +51 -0
  119. easy_rec/python/layers/mmoe.py +83 -0
  120. easy_rec/python/layers/multihead_attention.py +162 -0
  121. easy_rec/python/layers/multihead_cross_attention.py +749 -0
  122. easy_rec/python/layers/senet.py +73 -0
  123. easy_rec/python/layers/seq_input_layer.py +134 -0
  124. easy_rec/python/layers/sequence_feature_layer.py +249 -0
  125. easy_rec/python/layers/uniter.py +301 -0
  126. easy_rec/python/layers/utils.py +248 -0
  127. easy_rec/python/layers/variational_dropout_layer.py +130 -0
  128. easy_rec/python/loss/__init__.py +0 -0
  129. easy_rec/python/loss/circle_loss.py +82 -0
  130. easy_rec/python/loss/contrastive_loss.py +79 -0
  131. easy_rec/python/loss/f1_reweight_loss.py +38 -0
  132. easy_rec/python/loss/focal_loss.py +93 -0
  133. easy_rec/python/loss/jrc_loss.py +128 -0
  134. easy_rec/python/loss/listwise_loss.py +161 -0
  135. easy_rec/python/loss/multi_similarity.py +68 -0
  136. easy_rec/python/loss/pairwise_loss.py +307 -0
  137. easy_rec/python/loss/softmax_loss_with_negative_mining.py +110 -0
  138. easy_rec/python/loss/zero_inflated_lognormal.py +76 -0
  139. easy_rec/python/main.py +878 -0
  140. easy_rec/python/model/__init__.py +0 -0
  141. easy_rec/python/model/autoint.py +73 -0
  142. easy_rec/python/model/cmbf.py +47 -0
  143. easy_rec/python/model/collaborative_metric_learning.py +182 -0
  144. easy_rec/python/model/custom_model.py +323 -0
  145. easy_rec/python/model/dat.py +138 -0
  146. easy_rec/python/model/dbmtl.py +116 -0
  147. easy_rec/python/model/dcn.py +70 -0
  148. easy_rec/python/model/deepfm.py +106 -0
  149. easy_rec/python/model/dlrm.py +73 -0
  150. easy_rec/python/model/dropoutnet.py +207 -0
  151. easy_rec/python/model/dssm.py +154 -0
  152. easy_rec/python/model/dssm_senet.py +143 -0
  153. easy_rec/python/model/dummy_model.py +48 -0
  154. easy_rec/python/model/easy_rec_estimator.py +739 -0
  155. easy_rec/python/model/easy_rec_model.py +467 -0
  156. easy_rec/python/model/esmm.py +242 -0
  157. easy_rec/python/model/fm.py +63 -0
  158. easy_rec/python/model/match_model.py +357 -0
  159. easy_rec/python/model/mind.py +445 -0
  160. easy_rec/python/model/mmoe.py +70 -0
  161. easy_rec/python/model/multi_task_model.py +303 -0
  162. easy_rec/python/model/multi_tower.py +62 -0
  163. easy_rec/python/model/multi_tower_bst.py +190 -0
  164. easy_rec/python/model/multi_tower_din.py +130 -0
  165. easy_rec/python/model/multi_tower_recall.py +68 -0
  166. easy_rec/python/model/pdn.py +203 -0
  167. easy_rec/python/model/ple.py +120 -0
  168. easy_rec/python/model/rank_model.py +485 -0
  169. easy_rec/python/model/rocket_launching.py +203 -0
  170. easy_rec/python/model/simple_multi_task.py +54 -0
  171. easy_rec/python/model/uniter.py +46 -0
  172. easy_rec/python/model/wide_and_deep.py +121 -0
  173. easy_rec/python/ops/1.12/incr_record.so +0 -0
  174. easy_rec/python/ops/1.12/kafka.so +0 -0
  175. easy_rec/python/ops/1.12/libcustom_ops.so +0 -0
  176. easy_rec/python/ops/1.12/libembed_op.so +0 -0
  177. easy_rec/python/ops/1.12/libhiredis.so.1.0.0 +0 -0
  178. easy_rec/python/ops/1.12/librdkafka++.so.1 +0 -0
  179. easy_rec/python/ops/1.12/librdkafka.so.1 +0 -0
  180. easy_rec/python/ops/1.12/libredis++.so +0 -0
  181. easy_rec/python/ops/1.12/libredis++.so.1 +0 -0
  182. easy_rec/python/ops/1.12/libredis++.so.1.2.3 +0 -0
  183. easy_rec/python/ops/1.12/libstr_avx_op.so +0 -0
  184. easy_rec/python/ops/1.12/libwrite_sparse_kv.so +0 -0
  185. easy_rec/python/ops/1.15/incr_record.so +0 -0
  186. easy_rec/python/ops/1.15/kafka.so +0 -0
  187. easy_rec/python/ops/1.15/libcustom_ops.so +0 -0
  188. easy_rec/python/ops/1.15/libembed_op.so +0 -0
  189. easy_rec/python/ops/1.15/libhiredis.so.1.0.0 +0 -0
  190. easy_rec/python/ops/1.15/librdkafka++.so +0 -0
  191. easy_rec/python/ops/1.15/librdkafka++.so.1 +0 -0
  192. easy_rec/python/ops/1.15/librdkafka.so +0 -0
  193. easy_rec/python/ops/1.15/librdkafka.so.1 +0 -0
  194. easy_rec/python/ops/1.15/libredis++.so.1 +0 -0
  195. easy_rec/python/ops/1.15/libstr_avx_op.so +0 -0
  196. easy_rec/python/ops/2.12/libcustom_ops.so +0 -0
  197. easy_rec/python/ops/2.12/libload_embed.so +0 -0
  198. easy_rec/python/ops/2.12/libstr_avx_op.so +0 -0
  199. easy_rec/python/ops/__init__.py +0 -0
  200. easy_rec/python/ops/gen_kafka_ops.py +193 -0
  201. easy_rec/python/ops/gen_str_avx_op.py +28 -0
  202. easy_rec/python/ops/incr_record.py +30 -0
  203. easy_rec/python/predict.py +170 -0
  204. easy_rec/python/protos/__init__.py +0 -0
  205. easy_rec/python/protos/autoint_pb2.py +122 -0
  206. easy_rec/python/protos/backbone_pb2.py +1416 -0
  207. easy_rec/python/protos/cmbf_pb2.py +435 -0
  208. easy_rec/python/protos/collaborative_metric_learning_pb2.py +252 -0
  209. easy_rec/python/protos/custom_model_pb2.py +57 -0
  210. easy_rec/python/protos/dat_pb2.py +262 -0
  211. easy_rec/python/protos/data_source_pb2.py +422 -0
  212. easy_rec/python/protos/dataset_pb2.py +1920 -0
  213. easy_rec/python/protos/dbmtl_pb2.py +191 -0
  214. easy_rec/python/protos/dcn_pb2.py +197 -0
  215. easy_rec/python/protos/deepfm_pb2.py +163 -0
  216. easy_rec/python/protos/dlrm_pb2.py +163 -0
  217. easy_rec/python/protos/dnn_pb2.py +329 -0
  218. easy_rec/python/protos/dropoutnet_pb2.py +239 -0
  219. easy_rec/python/protos/dssm_pb2.py +262 -0
  220. easy_rec/python/protos/dssm_senet_pb2.py +282 -0
  221. easy_rec/python/protos/easy_rec_model_pb2.py +1672 -0
  222. easy_rec/python/protos/esmm_pb2.py +133 -0
  223. easy_rec/python/protos/eval_pb2.py +930 -0
  224. easy_rec/python/protos/export_pb2.py +379 -0
  225. easy_rec/python/protos/feature_config_pb2.py +1359 -0
  226. easy_rec/python/protos/fm_pb2.py +90 -0
  227. easy_rec/python/protos/hive_config_pb2.py +138 -0
  228. easy_rec/python/protos/hyperparams_pb2.py +624 -0
  229. easy_rec/python/protos/keras_layer_pb2.py +692 -0
  230. easy_rec/python/protos/layer_pb2.py +1936 -0
  231. easy_rec/python/protos/loss_pb2.py +1713 -0
  232. easy_rec/python/protos/mind_pb2.py +497 -0
  233. easy_rec/python/protos/mmoe_pb2.py +215 -0
  234. easy_rec/python/protos/multi_tower_pb2.py +295 -0
  235. easy_rec/python/protos/multi_tower_recall_pb2.py +198 -0
  236. easy_rec/python/protos/optimizer_pb2.py +2017 -0
  237. easy_rec/python/protos/pdn_pb2.py +293 -0
  238. easy_rec/python/protos/pipeline_pb2.py +516 -0
  239. easy_rec/python/protos/ple_pb2.py +231 -0
  240. easy_rec/python/protos/predict_pb2.py +1140 -0
  241. easy_rec/python/protos/rocket_launching_pb2.py +169 -0
  242. easy_rec/python/protos/seq_encoder_pb2.py +1084 -0
  243. easy_rec/python/protos/simi_pb2.py +54 -0
  244. easy_rec/python/protos/simple_multi_task_pb2.py +97 -0
  245. easy_rec/python/protos/tf_predict_pb2.py +630 -0
  246. easy_rec/python/protos/tower_pb2.py +661 -0
  247. easy_rec/python/protos/train_pb2.py +1197 -0
  248. easy_rec/python/protos/uniter_pb2.py +307 -0
  249. easy_rec/python/protos/variational_dropout_pb2.py +91 -0
  250. easy_rec/python/protos/wide_and_deep_pb2.py +131 -0
  251. easy_rec/python/test/__init__.py +0 -0
  252. easy_rec/python/test/csv_input_test.py +340 -0
  253. easy_rec/python/test/custom_early_stop_func.py +19 -0
  254. easy_rec/python/test/dh_local_run.py +104 -0
  255. easy_rec/python/test/embed_test.py +155 -0
  256. easy_rec/python/test/emr_run.py +119 -0
  257. easy_rec/python/test/eval_metric_test.py +107 -0
  258. easy_rec/python/test/excel_convert_test.py +64 -0
  259. easy_rec/python/test/export_test.py +513 -0
  260. easy_rec/python/test/fg_test.py +70 -0
  261. easy_rec/python/test/hive_input_test.py +311 -0
  262. easy_rec/python/test/hpo_test.py +235 -0
  263. easy_rec/python/test/kafka_test.py +373 -0
  264. easy_rec/python/test/local_incr_test.py +122 -0
  265. easy_rec/python/test/loss_test.py +110 -0
  266. easy_rec/python/test/odps_command.py +61 -0
  267. easy_rec/python/test/odps_local_run.py +86 -0
  268. easy_rec/python/test/odps_run.py +254 -0
  269. easy_rec/python/test/odps_test_cls.py +39 -0
  270. easy_rec/python/test/odps_test_prepare.py +198 -0
  271. easy_rec/python/test/odps_test_util.py +237 -0
  272. easy_rec/python/test/pre_check_test.py +54 -0
  273. easy_rec/python/test/predictor_test.py +394 -0
  274. easy_rec/python/test/rtp_convert_test.py +133 -0
  275. easy_rec/python/test/run.py +138 -0
  276. easy_rec/python/test/train_eval_test.py +1299 -0
  277. easy_rec/python/test/util_test.py +85 -0
  278. easy_rec/python/test/zero_inflated_lognormal_test.py +53 -0
  279. easy_rec/python/tools/__init__.py +0 -0
  280. easy_rec/python/tools/add_boundaries_to_config.py +67 -0
  281. easy_rec/python/tools/add_feature_info_to_config.py +145 -0
  282. easy_rec/python/tools/convert_config_format.py +48 -0
  283. easy_rec/python/tools/convert_rtp_data.py +79 -0
  284. easy_rec/python/tools/convert_rtp_fg.py +106 -0
  285. easy_rec/python/tools/create_config_from_excel.py +427 -0
  286. easy_rec/python/tools/criteo/__init__.py +0 -0
  287. easy_rec/python/tools/criteo/convert_data.py +157 -0
  288. easy_rec/python/tools/edit_lookup_graph.py +134 -0
  289. easy_rec/python/tools/faiss_index_pai.py +116 -0
  290. easy_rec/python/tools/feature_selection.py +316 -0
  291. easy_rec/python/tools/hit_rate_ds.py +223 -0
  292. easy_rec/python/tools/hit_rate_pai.py +138 -0
  293. easy_rec/python/tools/pre_check.py +120 -0
  294. easy_rec/python/tools/predict_and_chk.py +111 -0
  295. easy_rec/python/tools/read_kafka.py +55 -0
  296. easy_rec/python/tools/split_model_pai.py +286 -0
  297. easy_rec/python/tools/split_pdn_model_pai.py +272 -0
  298. easy_rec/python/tools/test_saved_model.py +80 -0
  299. easy_rec/python/tools/view_saved_model.py +39 -0
  300. easy_rec/python/tools/write_kafka.py +65 -0
  301. easy_rec/python/train_eval.py +325 -0
  302. easy_rec/python/utils/__init__.py +15 -0
  303. easy_rec/python/utils/activation.py +120 -0
  304. easy_rec/python/utils/check_utils.py +87 -0
  305. easy_rec/python/utils/compat.py +14 -0
  306. easy_rec/python/utils/config_util.py +652 -0
  307. easy_rec/python/utils/constant.py +43 -0
  308. easy_rec/python/utils/convert_rtp_fg.py +616 -0
  309. easy_rec/python/utils/dag.py +192 -0
  310. easy_rec/python/utils/distribution_utils.py +268 -0
  311. easy_rec/python/utils/ds_util.py +65 -0
  312. easy_rec/python/utils/embedding_utils.py +73 -0
  313. easy_rec/python/utils/estimator_utils.py +1036 -0
  314. easy_rec/python/utils/export_big_model.py +630 -0
  315. easy_rec/python/utils/expr_util.py +118 -0
  316. easy_rec/python/utils/fg_util.py +53 -0
  317. easy_rec/python/utils/hit_rate_utils.py +220 -0
  318. easy_rec/python/utils/hive_utils.py +183 -0
  319. easy_rec/python/utils/hpo_util.py +137 -0
  320. easy_rec/python/utils/hvd_utils.py +56 -0
  321. easy_rec/python/utils/input_utils.py +108 -0
  322. easy_rec/python/utils/io_util.py +282 -0
  323. easy_rec/python/utils/load_class.py +249 -0
  324. easy_rec/python/utils/meta_graph_editor.py +941 -0
  325. easy_rec/python/utils/multi_optimizer.py +62 -0
  326. easy_rec/python/utils/numpy_utils.py +18 -0
  327. easy_rec/python/utils/odps_util.py +79 -0
  328. easy_rec/python/utils/pai_util.py +86 -0
  329. easy_rec/python/utils/proto_util.py +90 -0
  330. easy_rec/python/utils/restore_filter.py +89 -0
  331. easy_rec/python/utils/shape_utils.py +432 -0
  332. easy_rec/python/utils/static_shape.py +71 -0
  333. easy_rec/python/utils/test_utils.py +866 -0
  334. easy_rec/python/utils/tf_utils.py +56 -0
  335. easy_rec/version.py +4 -0
  336. test/__init__.py +0 -0
@@ -0,0 +1,497 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/mind.proto
3
+
4
+ import sys
5
+
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pb2
8
+ from google.protobuf import message as _message
9
+ from google.protobuf import reflection as _reflection
10
+ from google.protobuf import symbol_database as _symbol_database
11
+
12
+ from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2 # NOQA
13
+ from easy_rec.python.protos import simi_pb2 as easy__rec_dot_python_dot_protos_dot_simi__pb2 # NOQA
14
+
15
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
16
+ # @@protoc_insertion_point(imports)
17
+
18
+ _sym_db = _symbol_database.Default()
19
+
20
+ DESCRIPTOR = _descriptor.FileDescriptor(
21
+ name='easy_rec/python/protos/mind.proto',
22
+ package='protos',
23
+ syntax='proto2',
24
+ serialized_pb=_b(
25
+ '\n!easy_rec/python/protos/mind.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\x1a!easy_rec/python/protos/simi.proto\"\xea\x01\n\x07\x43\x61psule\x12\x10\n\x05max_k\x18\x01 \x01(\r:\x01\x35\x12\x13\n\x0bmax_seq_len\x18\x02 \x02(\r\x12\x10\n\x08high_dim\x18\x03 \x02(\r\x12\x14\n\tnum_iters\x18\x04 \x01(\r:\x01\x33\x12 \n\x14routing_logits_scale\x18\x05 \x01(\x02:\x02\x32\x30\x12 \n\x15routing_logits_stddev\x18\x06 \x01(\x02:\x01\x31\x12\x15\n\nsquash_pow\x18\x07 \x01(\x02:\x01\x31\x12\x16\n\x0bscale_ratio\x18\x08 \x01(\x02:\x01\x31\x12\x1d\n\x0e\x63onst_caps_num\x18\t \x01(\x08:\x05\x66\x61lse\"\x92\x04\n\x04MIND\x12$\n\x0fpre_capsule_dnn\x18\x65 \x01(\x0b\x32\x0b.protos.DNN\x12\x1d\n\x08user_dnn\x18\x66 \x02(\x0b\x32\x0b.protos.DNN\x12\x1f\n\nconcat_dnn\x18g \x02(\x0b\x32\x0b.protos.DNN\x12@\n\x10user_seq_combine\x18h \x01(\x0e\x32!.protos.MIND.UserSeqCombineMethod:\x03SUM\x12\x1d\n\x08item_dnn\x18\x02 \x02(\x0b\x32\x0b.protos.DNN\x12\'\n\x0e\x63\x61psule_config\x18\x03 \x02(\x0b\x32\x0f.protos.Capsule\x12\x14\n\x08simi_pow\x18\x04 \x01(\x02:\x02\x31\x30\x12-\n\tsimi_func\x18\x05 \x01(\x0e\x32\x12.protos.Similarity:\x06\x43OSINE\x12\x18\n\nscale_simi\x18\x06 \x01(\x08:\x04true\x12!\n\x11l2_regularization\x18\x07 \x02(\x02:\x06\x30.0001\x12\x13\n\x0btime_id_fea\x18\x08 \x01(\t\x12\x0f\n\x07item_id\x18\t \x01(\t\x12&\n\x17ignore_in_batch_neg_sam\x18\n \x01(\x08:\x05\x66\x61lse\x12\x1d\n\x12max_interests_simi\x18\x0b \x01(\x02:\x01\x31\"+\n\x14UserSeqCombineMethod\x12\n\n\x06\x43ONCAT\x10\x00\x12\x07\n\x03SUM\x10\x01'
26
+ ),
27
+ dependencies=[
28
+ easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
29
+ easy__rec_dot_python_dot_protos_dot_simi__pb2.DESCRIPTOR,
30
+ ])
31
+
32
+ _MIND_USERSEQCOMBINEMETHOD = _descriptor.EnumDescriptor(
33
+ name='UserSeqCombineMethod',
34
+ full_name='protos.MIND.UserSeqCombineMethod',
35
+ filename=None,
36
+ file=DESCRIPTOR,
37
+ values=[
38
+ _descriptor.EnumValueDescriptor(
39
+ name='CONCAT', index=0, number=0, options=None, type=None),
40
+ _descriptor.EnumValueDescriptor(
41
+ name='SUM', index=1, number=1, options=None, type=None),
42
+ ],
43
+ containing_type=None,
44
+ options=None,
45
+ serialized_start=839,
46
+ serialized_end=882,
47
+ )
48
+ _sym_db.RegisterEnumDescriptor(_MIND_USERSEQCOMBINEMETHOD)
49
+
50
+ _CAPSULE = _descriptor.Descriptor(
51
+ name='Capsule',
52
+ full_name='protos.Capsule',
53
+ filename=None,
54
+ file=DESCRIPTOR,
55
+ containing_type=None,
56
+ fields=[
57
+ _descriptor.FieldDescriptor(
58
+ name='max_k',
59
+ full_name='protos.Capsule.max_k',
60
+ index=0,
61
+ number=1,
62
+ type=13,
63
+ cpp_type=3,
64
+ label=1,
65
+ has_default_value=True,
66
+ default_value=5,
67
+ message_type=None,
68
+ enum_type=None,
69
+ containing_type=None,
70
+ is_extension=False,
71
+ extension_scope=None,
72
+ options=None),
73
+ _descriptor.FieldDescriptor(
74
+ name='max_seq_len',
75
+ full_name='protos.Capsule.max_seq_len',
76
+ index=1,
77
+ number=2,
78
+ type=13,
79
+ cpp_type=3,
80
+ label=2,
81
+ has_default_value=False,
82
+ default_value=0,
83
+ message_type=None,
84
+ enum_type=None,
85
+ containing_type=None,
86
+ is_extension=False,
87
+ extension_scope=None,
88
+ options=None),
89
+ _descriptor.FieldDescriptor(
90
+ name='high_dim',
91
+ full_name='protos.Capsule.high_dim',
92
+ index=2,
93
+ number=3,
94
+ type=13,
95
+ cpp_type=3,
96
+ label=2,
97
+ has_default_value=False,
98
+ default_value=0,
99
+ message_type=None,
100
+ enum_type=None,
101
+ containing_type=None,
102
+ is_extension=False,
103
+ extension_scope=None,
104
+ options=None),
105
+ _descriptor.FieldDescriptor(
106
+ name='num_iters',
107
+ full_name='protos.Capsule.num_iters',
108
+ index=3,
109
+ number=4,
110
+ type=13,
111
+ cpp_type=3,
112
+ label=1,
113
+ has_default_value=True,
114
+ default_value=3,
115
+ message_type=None,
116
+ enum_type=None,
117
+ containing_type=None,
118
+ is_extension=False,
119
+ extension_scope=None,
120
+ options=None),
121
+ _descriptor.FieldDescriptor(
122
+ name='routing_logits_scale',
123
+ full_name='protos.Capsule.routing_logits_scale',
124
+ index=4,
125
+ number=5,
126
+ type=2,
127
+ cpp_type=6,
128
+ label=1,
129
+ has_default_value=True,
130
+ default_value=float(20),
131
+ message_type=None,
132
+ enum_type=None,
133
+ containing_type=None,
134
+ is_extension=False,
135
+ extension_scope=None,
136
+ options=None),
137
+ _descriptor.FieldDescriptor(
138
+ name='routing_logits_stddev',
139
+ full_name='protos.Capsule.routing_logits_stddev',
140
+ index=5,
141
+ number=6,
142
+ type=2,
143
+ cpp_type=6,
144
+ label=1,
145
+ has_default_value=True,
146
+ default_value=float(1),
147
+ message_type=None,
148
+ enum_type=None,
149
+ containing_type=None,
150
+ is_extension=False,
151
+ extension_scope=None,
152
+ options=None),
153
+ _descriptor.FieldDescriptor(
154
+ name='squash_pow',
155
+ full_name='protos.Capsule.squash_pow',
156
+ index=6,
157
+ number=7,
158
+ type=2,
159
+ cpp_type=6,
160
+ label=1,
161
+ has_default_value=True,
162
+ default_value=float(1),
163
+ message_type=None,
164
+ enum_type=None,
165
+ containing_type=None,
166
+ is_extension=False,
167
+ extension_scope=None,
168
+ options=None),
169
+ _descriptor.FieldDescriptor(
170
+ name='scale_ratio',
171
+ full_name='protos.Capsule.scale_ratio',
172
+ index=7,
173
+ number=8,
174
+ type=2,
175
+ cpp_type=6,
176
+ label=1,
177
+ has_default_value=True,
178
+ default_value=float(1),
179
+ message_type=None,
180
+ enum_type=None,
181
+ containing_type=None,
182
+ is_extension=False,
183
+ extension_scope=None,
184
+ options=None),
185
+ _descriptor.FieldDescriptor(
186
+ name='const_caps_num',
187
+ full_name='protos.Capsule.const_caps_num',
188
+ index=8,
189
+ number=9,
190
+ type=8,
191
+ cpp_type=7,
192
+ label=1,
193
+ has_default_value=True,
194
+ default_value=False,
195
+ message_type=None,
196
+ enum_type=None,
197
+ containing_type=None,
198
+ is_extension=False,
199
+ extension_scope=None,
200
+ options=None),
201
+ ],
202
+ extensions=[],
203
+ nested_types=[],
204
+ enum_types=[],
205
+ options=None,
206
+ is_extendable=False,
207
+ syntax='proto2',
208
+ extension_ranges=[],
209
+ oneofs=[],
210
+ serialized_start=115,
211
+ serialized_end=349,
212
+ )
213
+
214
+ _MIND = _descriptor.Descriptor(
215
+ name='MIND',
216
+ full_name='protos.MIND',
217
+ filename=None,
218
+ file=DESCRIPTOR,
219
+ containing_type=None,
220
+ fields=[
221
+ _descriptor.FieldDescriptor(
222
+ name='pre_capsule_dnn',
223
+ full_name='protos.MIND.pre_capsule_dnn',
224
+ index=0,
225
+ number=101,
226
+ type=11,
227
+ cpp_type=10,
228
+ label=1,
229
+ has_default_value=False,
230
+ default_value=None,
231
+ message_type=None,
232
+ enum_type=None,
233
+ containing_type=None,
234
+ is_extension=False,
235
+ extension_scope=None,
236
+ options=None),
237
+ _descriptor.FieldDescriptor(
238
+ name='user_dnn',
239
+ full_name='protos.MIND.user_dnn',
240
+ index=1,
241
+ number=102,
242
+ type=11,
243
+ cpp_type=10,
244
+ label=2,
245
+ has_default_value=False,
246
+ default_value=None,
247
+ message_type=None,
248
+ enum_type=None,
249
+ containing_type=None,
250
+ is_extension=False,
251
+ extension_scope=None,
252
+ options=None),
253
+ _descriptor.FieldDescriptor(
254
+ name='concat_dnn',
255
+ full_name='protos.MIND.concat_dnn',
256
+ index=2,
257
+ number=103,
258
+ type=11,
259
+ cpp_type=10,
260
+ label=2,
261
+ has_default_value=False,
262
+ default_value=None,
263
+ message_type=None,
264
+ enum_type=None,
265
+ containing_type=None,
266
+ is_extension=False,
267
+ extension_scope=None,
268
+ options=None),
269
+ _descriptor.FieldDescriptor(
270
+ name='user_seq_combine',
271
+ full_name='protos.MIND.user_seq_combine',
272
+ index=3,
273
+ number=104,
274
+ type=14,
275
+ cpp_type=8,
276
+ label=1,
277
+ has_default_value=True,
278
+ default_value=1,
279
+ message_type=None,
280
+ enum_type=None,
281
+ containing_type=None,
282
+ is_extension=False,
283
+ extension_scope=None,
284
+ options=None),
285
+ _descriptor.FieldDescriptor(
286
+ name='item_dnn',
287
+ full_name='protos.MIND.item_dnn',
288
+ index=4,
289
+ number=2,
290
+ type=11,
291
+ cpp_type=10,
292
+ label=2,
293
+ has_default_value=False,
294
+ default_value=None,
295
+ message_type=None,
296
+ enum_type=None,
297
+ containing_type=None,
298
+ is_extension=False,
299
+ extension_scope=None,
300
+ options=None),
301
+ _descriptor.FieldDescriptor(
302
+ name='capsule_config',
303
+ full_name='protos.MIND.capsule_config',
304
+ index=5,
305
+ number=3,
306
+ type=11,
307
+ cpp_type=10,
308
+ label=2,
309
+ has_default_value=False,
310
+ default_value=None,
311
+ message_type=None,
312
+ enum_type=None,
313
+ containing_type=None,
314
+ is_extension=False,
315
+ extension_scope=None,
316
+ options=None),
317
+ _descriptor.FieldDescriptor(
318
+ name='simi_pow',
319
+ full_name='protos.MIND.simi_pow',
320
+ index=6,
321
+ number=4,
322
+ type=2,
323
+ cpp_type=6,
324
+ label=1,
325
+ has_default_value=True,
326
+ default_value=float(10),
327
+ message_type=None,
328
+ enum_type=None,
329
+ containing_type=None,
330
+ is_extension=False,
331
+ extension_scope=None,
332
+ options=None),
333
+ _descriptor.FieldDescriptor(
334
+ name='simi_func',
335
+ full_name='protos.MIND.simi_func',
336
+ index=7,
337
+ number=5,
338
+ type=14,
339
+ cpp_type=8,
340
+ label=1,
341
+ has_default_value=True,
342
+ default_value=0,
343
+ message_type=None,
344
+ enum_type=None,
345
+ containing_type=None,
346
+ is_extension=False,
347
+ extension_scope=None,
348
+ options=None),
349
+ _descriptor.FieldDescriptor(
350
+ name='scale_simi',
351
+ full_name='protos.MIND.scale_simi',
352
+ index=8,
353
+ number=6,
354
+ type=8,
355
+ cpp_type=7,
356
+ label=1,
357
+ has_default_value=True,
358
+ default_value=True,
359
+ message_type=None,
360
+ enum_type=None,
361
+ containing_type=None,
362
+ is_extension=False,
363
+ extension_scope=None,
364
+ options=None),
365
+ _descriptor.FieldDescriptor(
366
+ name='l2_regularization',
367
+ full_name='protos.MIND.l2_regularization',
368
+ index=9,
369
+ number=7,
370
+ type=2,
371
+ cpp_type=6,
372
+ label=2,
373
+ has_default_value=True,
374
+ default_value=float(0.0001),
375
+ message_type=None,
376
+ enum_type=None,
377
+ containing_type=None,
378
+ is_extension=False,
379
+ extension_scope=None,
380
+ options=None),
381
+ _descriptor.FieldDescriptor(
382
+ name='time_id_fea',
383
+ full_name='protos.MIND.time_id_fea',
384
+ index=10,
385
+ number=8,
386
+ type=9,
387
+ cpp_type=9,
388
+ label=1,
389
+ has_default_value=False,
390
+ default_value=_b('').decode('utf-8'),
391
+ message_type=None,
392
+ enum_type=None,
393
+ containing_type=None,
394
+ is_extension=False,
395
+ extension_scope=None,
396
+ options=None),
397
+ _descriptor.FieldDescriptor(
398
+ name='item_id',
399
+ full_name='protos.MIND.item_id',
400
+ index=11,
401
+ number=9,
402
+ type=9,
403
+ cpp_type=9,
404
+ label=1,
405
+ has_default_value=False,
406
+ default_value=_b('').decode('utf-8'),
407
+ message_type=None,
408
+ enum_type=None,
409
+ containing_type=None,
410
+ is_extension=False,
411
+ extension_scope=None,
412
+ options=None),
413
+ _descriptor.FieldDescriptor(
414
+ name='ignore_in_batch_neg_sam',
415
+ full_name='protos.MIND.ignore_in_batch_neg_sam',
416
+ index=12,
417
+ number=10,
418
+ type=8,
419
+ cpp_type=7,
420
+ label=1,
421
+ has_default_value=True,
422
+ default_value=False,
423
+ message_type=None,
424
+ enum_type=None,
425
+ containing_type=None,
426
+ is_extension=False,
427
+ extension_scope=None,
428
+ options=None),
429
+ _descriptor.FieldDescriptor(
430
+ name='max_interests_simi',
431
+ full_name='protos.MIND.max_interests_simi',
432
+ index=13,
433
+ number=11,
434
+ type=2,
435
+ cpp_type=6,
436
+ label=1,
437
+ has_default_value=True,
438
+ default_value=float(1),
439
+ message_type=None,
440
+ enum_type=None,
441
+ containing_type=None,
442
+ is_extension=False,
443
+ extension_scope=None,
444
+ options=None),
445
+ ],
446
+ extensions=[],
447
+ nested_types=[],
448
+ enum_types=[
449
+ _MIND_USERSEQCOMBINEMETHOD,
450
+ ],
451
+ options=None,
452
+ is_extendable=False,
453
+ syntax='proto2',
454
+ extension_ranges=[],
455
+ oneofs=[],
456
+ serialized_start=352,
457
+ serialized_end=882,
458
+ )
459
+
460
+ _MIND.fields_by_name[
461
+ 'pre_capsule_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
462
+ _MIND.fields_by_name[
463
+ 'user_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
464
+ _MIND.fields_by_name[
465
+ 'concat_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
466
+ _MIND.fields_by_name['user_seq_combine'].enum_type = _MIND_USERSEQCOMBINEMETHOD
467
+ _MIND.fields_by_name[
468
+ 'item_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
469
+ _MIND.fields_by_name['capsule_config'].message_type = _CAPSULE
470
+ _MIND.fields_by_name[
471
+ 'simi_func'].enum_type = easy__rec_dot_python_dot_protos_dot_simi__pb2._SIMILARITY
472
+ _MIND_USERSEQCOMBINEMETHOD.containing_type = _MIND
473
+ DESCRIPTOR.message_types_by_name['Capsule'] = _CAPSULE
474
+ DESCRIPTOR.message_types_by_name['MIND'] = _MIND
475
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
476
+
477
+ Capsule = _reflection.GeneratedProtocolMessageType(
478
+ 'Capsule',
479
+ (_message.Message,),
480
+ dict(
481
+ DESCRIPTOR=_CAPSULE,
482
+ __module__='easy_rec.python.protos.mind_pb2'
483
+ # @@protoc_insertion_point(class_scope:protos.Capsule)
484
+ ))
485
+ _sym_db.RegisterMessage(Capsule)
486
+
487
+ MIND = _reflection.GeneratedProtocolMessageType(
488
+ 'MIND',
489
+ (_message.Message,),
490
+ dict(
491
+ DESCRIPTOR=_MIND,
492
+ __module__='easy_rec.python.protos.mind_pb2'
493
+ # @@protoc_insertion_point(class_scope:protos.MIND)
494
+ ))
495
+ _sym_db.RegisterMessage(MIND)
496
+
497
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,215 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/mmoe.proto
3
+
4
+ import sys
5
+
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pb2
8
+ from google.protobuf import message as _message
9
+ from google.protobuf import reflection as _reflection
10
+ from google.protobuf import symbol_database as _symbol_database
11
+
12
+ from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2 # NOQA
13
+ from easy_rec.python.protos import tower_pb2 as easy__rec_dot_python_dot_protos_dot_tower__pb2 # NOQA
14
+
15
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
16
+ # @@protoc_insertion_point(imports)
17
+
18
+ _sym_db = _symbol_database.Default()
19
+
20
+ DESCRIPTOR = _descriptor.FileDescriptor(
21
+ name='easy_rec/python/protos/mmoe.proto',
22
+ package='protos',
23
+ syntax='proto2',
24
+ serialized_pb=_b(
25
+ '\n!easy_rec/python/protos/mmoe.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\x1a\"easy_rec/python/protos/tower.proto\"<\n\x0b\x45xpertTower\x12\x13\n\x0b\x65xpert_name\x18\x01 \x02(\t\x12\x18\n\x03\x64nn\x18\x02 \x02(\x0b\x32\x0b.protos.DNN\"\xaf\x01\n\x04MMoE\x12$\n\x07\x65xperts\x18\x01 \x03(\x0b\x32\x13.protos.ExpertTower\x12\x1f\n\nexpert_dnn\x18\x02 \x01(\x0b\x32\x0b.protos.DNN\x12\x15\n\nnum_expert\x18\x03 \x01(\r:\x01\x30\x12&\n\x0btask_towers\x18\x04 \x03(\x0b\x32\x11.protos.TaskTower\x12!\n\x11l2_regularization\x18\x05 \x02(\x02:\x06\x30.0001'
26
+ ),
27
+ dependencies=[
28
+ easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
29
+ easy__rec_dot_python_dot_protos_dot_tower__pb2.DESCRIPTOR,
30
+ ])
31
+
32
+ _EXPERTTOWER = _descriptor.Descriptor(
33
+ name='ExpertTower',
34
+ full_name='protos.ExpertTower',
35
+ filename=None,
36
+ file=DESCRIPTOR,
37
+ containing_type=None,
38
+ fields=[
39
+ _descriptor.FieldDescriptor(
40
+ name='expert_name',
41
+ full_name='protos.ExpertTower.expert_name',
42
+ index=0,
43
+ number=1,
44
+ type=9,
45
+ cpp_type=9,
46
+ label=2,
47
+ has_default_value=False,
48
+ default_value=_b('').decode('utf-8'),
49
+ message_type=None,
50
+ enum_type=None,
51
+ containing_type=None,
52
+ is_extension=False,
53
+ extension_scope=None,
54
+ options=None),
55
+ _descriptor.FieldDescriptor(
56
+ name='dnn',
57
+ full_name='protos.ExpertTower.dnn',
58
+ index=1,
59
+ number=2,
60
+ type=11,
61
+ cpp_type=10,
62
+ label=2,
63
+ has_default_value=False,
64
+ default_value=None,
65
+ message_type=None,
66
+ enum_type=None,
67
+ containing_type=None,
68
+ is_extension=False,
69
+ extension_scope=None,
70
+ options=None),
71
+ ],
72
+ extensions=[],
73
+ nested_types=[],
74
+ enum_types=[],
75
+ options=None,
76
+ is_extendable=False,
77
+ syntax='proto2',
78
+ extension_ranges=[],
79
+ oneofs=[],
80
+ serialized_start=115,
81
+ serialized_end=175,
82
+ )
83
+
84
+ _MMOE = _descriptor.Descriptor(
85
+ name='MMoE',
86
+ full_name='protos.MMoE',
87
+ filename=None,
88
+ file=DESCRIPTOR,
89
+ containing_type=None,
90
+ fields=[
91
+ _descriptor.FieldDescriptor(
92
+ name='experts',
93
+ full_name='protos.MMoE.experts',
94
+ index=0,
95
+ number=1,
96
+ type=11,
97
+ cpp_type=10,
98
+ label=3,
99
+ has_default_value=False,
100
+ default_value=[],
101
+ message_type=None,
102
+ enum_type=None,
103
+ containing_type=None,
104
+ is_extension=False,
105
+ extension_scope=None,
106
+ options=None),
107
+ _descriptor.FieldDescriptor(
108
+ name='expert_dnn',
109
+ full_name='protos.MMoE.expert_dnn',
110
+ index=1,
111
+ number=2,
112
+ type=11,
113
+ cpp_type=10,
114
+ label=1,
115
+ has_default_value=False,
116
+ default_value=None,
117
+ message_type=None,
118
+ enum_type=None,
119
+ containing_type=None,
120
+ is_extension=False,
121
+ extension_scope=None,
122
+ options=None),
123
+ _descriptor.FieldDescriptor(
124
+ name='num_expert',
125
+ full_name='protos.MMoE.num_expert',
126
+ index=2,
127
+ number=3,
128
+ type=13,
129
+ cpp_type=3,
130
+ label=1,
131
+ has_default_value=True,
132
+ default_value=0,
133
+ message_type=None,
134
+ enum_type=None,
135
+ containing_type=None,
136
+ is_extension=False,
137
+ extension_scope=None,
138
+ options=None),
139
+ _descriptor.FieldDescriptor(
140
+ name='task_towers',
141
+ full_name='protos.MMoE.task_towers',
142
+ index=3,
143
+ number=4,
144
+ type=11,
145
+ cpp_type=10,
146
+ label=3,
147
+ has_default_value=False,
148
+ default_value=[],
149
+ message_type=None,
150
+ enum_type=None,
151
+ containing_type=None,
152
+ is_extension=False,
153
+ extension_scope=None,
154
+ options=None),
155
+ _descriptor.FieldDescriptor(
156
+ name='l2_regularization',
157
+ full_name='protos.MMoE.l2_regularization',
158
+ index=4,
159
+ number=5,
160
+ type=2,
161
+ cpp_type=6,
162
+ label=2,
163
+ has_default_value=True,
164
+ default_value=float(0.0001),
165
+ message_type=None,
166
+ enum_type=None,
167
+ containing_type=None,
168
+ is_extension=False,
169
+ extension_scope=None,
170
+ options=None),
171
+ ],
172
+ extensions=[],
173
+ nested_types=[],
174
+ enum_types=[],
175
+ options=None,
176
+ is_extendable=False,
177
+ syntax='proto2',
178
+ extension_ranges=[],
179
+ oneofs=[],
180
+ serialized_start=178,
181
+ serialized_end=353,
182
+ )
183
+
184
+ _EXPERTTOWER.fields_by_name[
185
+ 'dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
186
+ _MMOE.fields_by_name['experts'].message_type = _EXPERTTOWER
187
+ _MMOE.fields_by_name[
188
+ 'expert_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
189
+ _MMOE.fields_by_name[
190
+ 'task_towers'].message_type = easy__rec_dot_python_dot_protos_dot_tower__pb2._TASKTOWER
191
+ DESCRIPTOR.message_types_by_name['ExpertTower'] = _EXPERTTOWER
192
+ DESCRIPTOR.message_types_by_name['MMoE'] = _MMOE
193
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
194
+
195
+ ExpertTower = _reflection.GeneratedProtocolMessageType(
196
+ 'ExpertTower',
197
+ (_message.Message,),
198
+ dict(
199
+ DESCRIPTOR=_EXPERTTOWER,
200
+ __module__='easy_rec.python.protos.mmoe_pb2'
201
+ # @@protoc_insertion_point(class_scope:protos.ExpertTower)
202
+ ))
203
+ _sym_db.RegisterMessage(ExpertTower)
204
+
205
+ MMoE = _reflection.GeneratedProtocolMessageType(
206
+ 'MMoE',
207
+ (_message.Message,),
208
+ dict(
209
+ DESCRIPTOR=_MMOE,
210
+ __module__='easy_rec.python.protos.mmoe_pb2'
211
+ # @@protoc_insertion_point(class_scope:protos.MMoE)
212
+ ))
213
+ _sym_db.RegisterMessage(MMoE)
214
+
215
+ # @@protoc_insertion_point(module_scope)