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,435 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/cmbf.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
+
14
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
15
+ # @@protoc_insertion_point(imports)
16
+
17
+ _sym_db = _symbol_database.Default()
18
+
19
+ DESCRIPTOR = _descriptor.FileDescriptor(
20
+ name='easy_rec/python/protos/cmbf.proto',
21
+ package='protos',
22
+ syntax='proto2',
23
+ serialized_pb=_b(
24
+ '\n!easy_rec/python/protos/cmbf.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\"\x95\x05\n\tCMBFTower\x12\x19\n\x0emulti_head_num\x18\x01 \x02(\r:\x01\x31\x12\x1f\n\x14image_multi_head_num\x18\x65 \x02(\r:\x01\x31\x12\x1e\n\x13text_multi_head_num\x18\x66 \x02(\r:\x01\x31\x12\x16\n\x0etext_head_size\x18\x02 \x02(\r\x12\x1b\n\x0fimage_head_size\x18\x03 \x02(\r:\x02\x36\x34\x12\"\n\x17image_feature_patch_num\x18\x04 \x02(\r:\x01\x31\x12\x1c\n\x11image_feature_dim\x18\x05 \x02(\r:\x01\x30\x12)\n\x1eimage_self_attention_layer_num\x18\x06 \x02(\r:\x01\x30\x12(\n\x1dtext_self_attention_layer_num\x18\x07 \x02(\r:\x01\x31\x12 \n\x15\x63ross_modal_layer_num\x18\x08 \x02(\r:\x01\x31\x12\x1d\n\x15image_cross_head_size\x18\t \x02(\r\x12\x1c\n\x14text_cross_head_size\x18\n \x02(\r\x12\x1e\n\x13hidden_dropout_prob\x18\x0b \x02(\x02:\x01\x30\x12\'\n\x1c\x61ttention_probs_dropout_prob\x18\x0c \x02(\x02:\x01\x30\x12\x1d\n\x0euse_token_type\x18\r \x02(\x08:\x05\x66\x61lse\x12%\n\x17use_position_embeddings\x18\x0e \x02(\x08:\x04true\x12\"\n\x17max_position_embeddings\x18\x0f \x02(\r:\x01\x30\x12&\n\x19text_seq_emb_dropout_prob\x18\x10 \x02(\x02:\x03\x30.1\x12&\n\x11other_feature_dnn\x18\x11 \x01(\x0b\x32\x0b.protos.DNN\"I\n\x04\x43MBF\x12!\n\x06\x63onfig\x18\x01 \x02(\x0b\x32\x11.protos.CMBFTower\x12\x1e\n\tfinal_dnn\x18\x02 \x02(\x0b\x32\x0b.protos.DNN'
25
+ ),
26
+ dependencies=[
27
+ easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
28
+ ])
29
+
30
+ _CMBFTOWER = _descriptor.Descriptor(
31
+ name='CMBFTower',
32
+ full_name='protos.CMBFTower',
33
+ filename=None,
34
+ file=DESCRIPTOR,
35
+ containing_type=None,
36
+ fields=[
37
+ _descriptor.FieldDescriptor(
38
+ name='multi_head_num',
39
+ full_name='protos.CMBFTower.multi_head_num',
40
+ index=0,
41
+ number=1,
42
+ type=13,
43
+ cpp_type=3,
44
+ label=2,
45
+ has_default_value=True,
46
+ default_value=1,
47
+ message_type=None,
48
+ enum_type=None,
49
+ containing_type=None,
50
+ is_extension=False,
51
+ extension_scope=None,
52
+ options=None),
53
+ _descriptor.FieldDescriptor(
54
+ name='image_multi_head_num',
55
+ full_name='protos.CMBFTower.image_multi_head_num',
56
+ index=1,
57
+ number=101,
58
+ type=13,
59
+ cpp_type=3,
60
+ label=2,
61
+ has_default_value=True,
62
+ default_value=1,
63
+ message_type=None,
64
+ enum_type=None,
65
+ containing_type=None,
66
+ is_extension=False,
67
+ extension_scope=None,
68
+ options=None),
69
+ _descriptor.FieldDescriptor(
70
+ name='text_multi_head_num',
71
+ full_name='protos.CMBFTower.text_multi_head_num',
72
+ index=2,
73
+ number=102,
74
+ type=13,
75
+ cpp_type=3,
76
+ label=2,
77
+ has_default_value=True,
78
+ default_value=1,
79
+ message_type=None,
80
+ enum_type=None,
81
+ containing_type=None,
82
+ is_extension=False,
83
+ extension_scope=None,
84
+ options=None),
85
+ _descriptor.FieldDescriptor(
86
+ name='text_head_size',
87
+ full_name='protos.CMBFTower.text_head_size',
88
+ index=3,
89
+ number=2,
90
+ type=13,
91
+ cpp_type=3,
92
+ label=2,
93
+ has_default_value=False,
94
+ default_value=0,
95
+ message_type=None,
96
+ enum_type=None,
97
+ containing_type=None,
98
+ is_extension=False,
99
+ extension_scope=None,
100
+ options=None),
101
+ _descriptor.FieldDescriptor(
102
+ name='image_head_size',
103
+ full_name='protos.CMBFTower.image_head_size',
104
+ index=4,
105
+ number=3,
106
+ type=13,
107
+ cpp_type=3,
108
+ label=2,
109
+ has_default_value=True,
110
+ default_value=64,
111
+ message_type=None,
112
+ enum_type=None,
113
+ containing_type=None,
114
+ is_extension=False,
115
+ extension_scope=None,
116
+ options=None),
117
+ _descriptor.FieldDescriptor(
118
+ name='image_feature_patch_num',
119
+ full_name='protos.CMBFTower.image_feature_patch_num',
120
+ index=5,
121
+ number=4,
122
+ type=13,
123
+ cpp_type=3,
124
+ label=2,
125
+ has_default_value=True,
126
+ default_value=1,
127
+ message_type=None,
128
+ enum_type=None,
129
+ containing_type=None,
130
+ is_extension=False,
131
+ extension_scope=None,
132
+ options=None),
133
+ _descriptor.FieldDescriptor(
134
+ name='image_feature_dim',
135
+ full_name='protos.CMBFTower.image_feature_dim',
136
+ index=6,
137
+ number=5,
138
+ type=13,
139
+ cpp_type=3,
140
+ label=2,
141
+ has_default_value=True,
142
+ default_value=0,
143
+ message_type=None,
144
+ enum_type=None,
145
+ containing_type=None,
146
+ is_extension=False,
147
+ extension_scope=None,
148
+ options=None),
149
+ _descriptor.FieldDescriptor(
150
+ name='image_self_attention_layer_num',
151
+ full_name='protos.CMBFTower.image_self_attention_layer_num',
152
+ index=7,
153
+ number=6,
154
+ type=13,
155
+ cpp_type=3,
156
+ label=2,
157
+ has_default_value=True,
158
+ default_value=0,
159
+ message_type=None,
160
+ enum_type=None,
161
+ containing_type=None,
162
+ is_extension=False,
163
+ extension_scope=None,
164
+ options=None),
165
+ _descriptor.FieldDescriptor(
166
+ name='text_self_attention_layer_num',
167
+ full_name='protos.CMBFTower.text_self_attention_layer_num',
168
+ index=8,
169
+ number=7,
170
+ type=13,
171
+ cpp_type=3,
172
+ label=2,
173
+ has_default_value=True,
174
+ default_value=1,
175
+ message_type=None,
176
+ enum_type=None,
177
+ containing_type=None,
178
+ is_extension=False,
179
+ extension_scope=None,
180
+ options=None),
181
+ _descriptor.FieldDescriptor(
182
+ name='cross_modal_layer_num',
183
+ full_name='protos.CMBFTower.cross_modal_layer_num',
184
+ index=9,
185
+ number=8,
186
+ type=13,
187
+ cpp_type=3,
188
+ label=2,
189
+ has_default_value=True,
190
+ default_value=1,
191
+ message_type=None,
192
+ enum_type=None,
193
+ containing_type=None,
194
+ is_extension=False,
195
+ extension_scope=None,
196
+ options=None),
197
+ _descriptor.FieldDescriptor(
198
+ name='image_cross_head_size',
199
+ full_name='protos.CMBFTower.image_cross_head_size',
200
+ index=10,
201
+ number=9,
202
+ type=13,
203
+ cpp_type=3,
204
+ label=2,
205
+ has_default_value=False,
206
+ default_value=0,
207
+ message_type=None,
208
+ enum_type=None,
209
+ containing_type=None,
210
+ is_extension=False,
211
+ extension_scope=None,
212
+ options=None),
213
+ _descriptor.FieldDescriptor(
214
+ name='text_cross_head_size',
215
+ full_name='protos.CMBFTower.text_cross_head_size',
216
+ index=11,
217
+ number=10,
218
+ type=13,
219
+ cpp_type=3,
220
+ label=2,
221
+ has_default_value=False,
222
+ default_value=0,
223
+ message_type=None,
224
+ enum_type=None,
225
+ containing_type=None,
226
+ is_extension=False,
227
+ extension_scope=None,
228
+ options=None),
229
+ _descriptor.FieldDescriptor(
230
+ name='hidden_dropout_prob',
231
+ full_name='protos.CMBFTower.hidden_dropout_prob',
232
+ index=12,
233
+ number=11,
234
+ type=2,
235
+ cpp_type=6,
236
+ label=2,
237
+ has_default_value=True,
238
+ default_value=float(0),
239
+ message_type=None,
240
+ enum_type=None,
241
+ containing_type=None,
242
+ is_extension=False,
243
+ extension_scope=None,
244
+ options=None),
245
+ _descriptor.FieldDescriptor(
246
+ name='attention_probs_dropout_prob',
247
+ full_name='protos.CMBFTower.attention_probs_dropout_prob',
248
+ index=13,
249
+ number=12,
250
+ type=2,
251
+ cpp_type=6,
252
+ label=2,
253
+ has_default_value=True,
254
+ default_value=float(0),
255
+ message_type=None,
256
+ enum_type=None,
257
+ containing_type=None,
258
+ is_extension=False,
259
+ extension_scope=None,
260
+ options=None),
261
+ _descriptor.FieldDescriptor(
262
+ name='use_token_type',
263
+ full_name='protos.CMBFTower.use_token_type',
264
+ index=14,
265
+ number=13,
266
+ type=8,
267
+ cpp_type=7,
268
+ label=2,
269
+ has_default_value=True,
270
+ default_value=False,
271
+ message_type=None,
272
+ enum_type=None,
273
+ containing_type=None,
274
+ is_extension=False,
275
+ extension_scope=None,
276
+ options=None),
277
+ _descriptor.FieldDescriptor(
278
+ name='use_position_embeddings',
279
+ full_name='protos.CMBFTower.use_position_embeddings',
280
+ index=15,
281
+ number=14,
282
+ type=8,
283
+ cpp_type=7,
284
+ label=2,
285
+ has_default_value=True,
286
+ default_value=True,
287
+ message_type=None,
288
+ enum_type=None,
289
+ containing_type=None,
290
+ is_extension=False,
291
+ extension_scope=None,
292
+ options=None),
293
+ _descriptor.FieldDescriptor(
294
+ name='max_position_embeddings',
295
+ full_name='protos.CMBFTower.max_position_embeddings',
296
+ index=16,
297
+ number=15,
298
+ type=13,
299
+ cpp_type=3,
300
+ label=2,
301
+ has_default_value=True,
302
+ default_value=0,
303
+ message_type=None,
304
+ enum_type=None,
305
+ containing_type=None,
306
+ is_extension=False,
307
+ extension_scope=None,
308
+ options=None),
309
+ _descriptor.FieldDescriptor(
310
+ name='text_seq_emb_dropout_prob',
311
+ full_name='protos.CMBFTower.text_seq_emb_dropout_prob',
312
+ index=17,
313
+ number=16,
314
+ type=2,
315
+ cpp_type=6,
316
+ label=2,
317
+ has_default_value=True,
318
+ default_value=float(0.1),
319
+ message_type=None,
320
+ enum_type=None,
321
+ containing_type=None,
322
+ is_extension=False,
323
+ extension_scope=None,
324
+ options=None),
325
+ _descriptor.FieldDescriptor(
326
+ name='other_feature_dnn',
327
+ full_name='protos.CMBFTower.other_feature_dnn',
328
+ index=18,
329
+ number=17,
330
+ type=11,
331
+ cpp_type=10,
332
+ label=1,
333
+ has_default_value=False,
334
+ default_value=None,
335
+ message_type=None,
336
+ enum_type=None,
337
+ containing_type=None,
338
+ is_extension=False,
339
+ extension_scope=None,
340
+ options=None),
341
+ ],
342
+ extensions=[],
343
+ nested_types=[],
344
+ enum_types=[],
345
+ options=None,
346
+ is_extendable=False,
347
+ syntax='proto2',
348
+ extension_ranges=[],
349
+ oneofs=[],
350
+ serialized_start=80,
351
+ serialized_end=741,
352
+ )
353
+
354
+ _CMBF = _descriptor.Descriptor(
355
+ name='CMBF',
356
+ full_name='protos.CMBF',
357
+ filename=None,
358
+ file=DESCRIPTOR,
359
+ containing_type=None,
360
+ fields=[
361
+ _descriptor.FieldDescriptor(
362
+ name='config',
363
+ full_name='protos.CMBF.config',
364
+ index=0,
365
+ number=1,
366
+ type=11,
367
+ cpp_type=10,
368
+ label=2,
369
+ has_default_value=False,
370
+ default_value=None,
371
+ message_type=None,
372
+ enum_type=None,
373
+ containing_type=None,
374
+ is_extension=False,
375
+ extension_scope=None,
376
+ options=None),
377
+ _descriptor.FieldDescriptor(
378
+ name='final_dnn',
379
+ full_name='protos.CMBF.final_dnn',
380
+ index=1,
381
+ number=2,
382
+ type=11,
383
+ cpp_type=10,
384
+ label=2,
385
+ has_default_value=False,
386
+ default_value=None,
387
+ message_type=None,
388
+ enum_type=None,
389
+ containing_type=None,
390
+ is_extension=False,
391
+ extension_scope=None,
392
+ options=None),
393
+ ],
394
+ extensions=[],
395
+ nested_types=[],
396
+ enum_types=[],
397
+ options=None,
398
+ is_extendable=False,
399
+ syntax='proto2',
400
+ extension_ranges=[],
401
+ oneofs=[],
402
+ serialized_start=743,
403
+ serialized_end=816,
404
+ )
405
+
406
+ _CMBFTOWER.fields_by_name[
407
+ 'other_feature_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
408
+ _CMBF.fields_by_name['config'].message_type = _CMBFTOWER
409
+ _CMBF.fields_by_name[
410
+ 'final_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
411
+ DESCRIPTOR.message_types_by_name['CMBFTower'] = _CMBFTOWER
412
+ DESCRIPTOR.message_types_by_name['CMBF'] = _CMBF
413
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
414
+
415
+ CMBFTower = _reflection.GeneratedProtocolMessageType(
416
+ 'CMBFTower',
417
+ (_message.Message,),
418
+ dict(
419
+ DESCRIPTOR=_CMBFTOWER,
420
+ __module__='easy_rec.python.protos.cmbf_pb2'
421
+ # @@protoc_insertion_point(class_scope:protos.CMBFTower)
422
+ ))
423
+ _sym_db.RegisterMessage(CMBFTower)
424
+
425
+ CMBF = _reflection.GeneratedProtocolMessageType(
426
+ 'CMBF',
427
+ (_message.Message,),
428
+ dict(
429
+ DESCRIPTOR=_CMBF,
430
+ __module__='easy_rec.python.protos.cmbf_pb2'
431
+ # @@protoc_insertion_point(class_scope:protos.CMBF)
432
+ ))
433
+ _sym_db.RegisterMessage(CMBF)
434
+
435
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,252 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/collaborative_metric_learning.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 layer_pb2 as easy__rec_dot_python_dot_protos_dot_layer__pb2 # NOQA
14
+ from easy_rec.python.protos import loss_pb2 as easy__rec_dot_python_dot_protos_dot_loss__pb2 # NOQA
15
+
16
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
17
+ # @@protoc_insertion_point(imports)
18
+
19
+ _sym_db = _symbol_database.Default()
20
+
21
+ DESCRIPTOR = _descriptor.FileDescriptor(
22
+ name='easy_rec/python/protos/collaborative_metric_learning.proto',
23
+ package='protos',
24
+ syntax='proto2',
25
+ serialized_pb=_b(
26
+ '\n:easy_rec/python/protos/collaborative_metric_learning.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\x1a\"easy_rec/python/protos/layer.proto\x1a!easy_rec/python/protos/loss.proto\"\xd9\x02\n\x13\x43oMetricLearningI2I\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12%\n\x07highway\x18\x02 \x03(\x0b\x32\x14.protos.HighWayTower\x12\r\n\x05input\x18\x04 \x01(\t\x12\x18\n\x03\x64nn\x18\x05 \x02(\x0b\x32\x0b.protos.DNN\x12!\n\x11l2_regularization\x18\x06 \x02(\x02:\x06\x30.0001\x12&\n\x18output_l2_normalized_emb\x18\x07 \x02(\x08:\x04true\x12\x11\n\tsample_id\x18\x08 \x01(\t\x12)\n\x0b\x63ircle_loss\x18\x65 \x01(\x0b\x32\x12.protos.CircleLossH\x00\x12<\n\x15multi_similarity_loss\x18\x66 \x01(\x0b\x32\x1b.protos.MultiSimilarityLossH\x00\x12\x0f\n\x07item_id\x18\t \x01(\tB\x06\n\x04loss'
27
+ ),
28
+ dependencies=[
29
+ easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
30
+ easy__rec_dot_python_dot_protos_dot_layer__pb2.DESCRIPTOR,
31
+ easy__rec_dot_python_dot_protos_dot_loss__pb2.DESCRIPTOR,
32
+ ])
33
+
34
+ _COMETRICLEARNINGI2I = _descriptor.Descriptor(
35
+ name='CoMetricLearningI2I',
36
+ full_name='protos.CoMetricLearningI2I',
37
+ filename=None,
38
+ file=DESCRIPTOR,
39
+ containing_type=None,
40
+ fields=[
41
+ _descriptor.FieldDescriptor(
42
+ name='session_id',
43
+ full_name='protos.CoMetricLearningI2I.session_id',
44
+ index=0,
45
+ number=1,
46
+ type=9,
47
+ cpp_type=9,
48
+ label=1,
49
+ has_default_value=False,
50
+ default_value=_b('').decode('utf-8'),
51
+ message_type=None,
52
+ enum_type=None,
53
+ containing_type=None,
54
+ is_extension=False,
55
+ extension_scope=None,
56
+ options=None),
57
+ _descriptor.FieldDescriptor(
58
+ name='highway',
59
+ full_name='protos.CoMetricLearningI2I.highway',
60
+ index=1,
61
+ number=2,
62
+ type=11,
63
+ cpp_type=10,
64
+ label=3,
65
+ has_default_value=False,
66
+ default_value=[],
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='input',
75
+ full_name='protos.CoMetricLearningI2I.input',
76
+ index=2,
77
+ number=4,
78
+ type=9,
79
+ cpp_type=9,
80
+ label=1,
81
+ has_default_value=False,
82
+ default_value=_b('').decode('utf-8'),
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='dnn',
91
+ full_name='protos.CoMetricLearningI2I.dnn',
92
+ index=3,
93
+ number=5,
94
+ type=11,
95
+ cpp_type=10,
96
+ label=2,
97
+ has_default_value=False,
98
+ default_value=None,
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='l2_regularization',
107
+ full_name='protos.CoMetricLearningI2I.l2_regularization',
108
+ index=4,
109
+ number=6,
110
+ type=2,
111
+ cpp_type=6,
112
+ label=2,
113
+ has_default_value=True,
114
+ default_value=float(0.0001),
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='output_l2_normalized_emb',
123
+ full_name='protos.CoMetricLearningI2I.output_l2_normalized_emb',
124
+ index=5,
125
+ number=7,
126
+ type=8,
127
+ cpp_type=7,
128
+ label=2,
129
+ has_default_value=True,
130
+ default_value=True,
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='sample_id',
139
+ full_name='protos.CoMetricLearningI2I.sample_id',
140
+ index=6,
141
+ number=8,
142
+ type=9,
143
+ cpp_type=9,
144
+ label=1,
145
+ has_default_value=False,
146
+ default_value=_b('').decode('utf-8'),
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='circle_loss',
155
+ full_name='protos.CoMetricLearningI2I.circle_loss',
156
+ index=7,
157
+ number=101,
158
+ type=11,
159
+ cpp_type=10,
160
+ label=1,
161
+ has_default_value=False,
162
+ default_value=None,
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='multi_similarity_loss',
171
+ full_name='protos.CoMetricLearningI2I.multi_similarity_loss',
172
+ index=8,
173
+ number=102,
174
+ type=11,
175
+ cpp_type=10,
176
+ label=1,
177
+ has_default_value=False,
178
+ default_value=None,
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='item_id',
187
+ full_name='protos.CoMetricLearningI2I.item_id',
188
+ index=9,
189
+ number=9,
190
+ type=9,
191
+ cpp_type=9,
192
+ label=1,
193
+ has_default_value=False,
194
+ default_value=_b('').decode('utf-8'),
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
+ _descriptor.OneofDescriptor(
211
+ name='loss',
212
+ full_name='protos.CoMetricLearningI2I.loss',
213
+ index=0,
214
+ containing_type=None,
215
+ fields=[]),
216
+ ],
217
+ serialized_start=176,
218
+ serialized_end=521,
219
+ )
220
+
221
+ _COMETRICLEARNINGI2I.fields_by_name[
222
+ 'highway'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._HIGHWAYTOWER
223
+ _COMETRICLEARNINGI2I.fields_by_name[
224
+ 'dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
225
+ _COMETRICLEARNINGI2I.fields_by_name[
226
+ 'circle_loss'].message_type = easy__rec_dot_python_dot_protos_dot_loss__pb2._CIRCLELOSS
227
+ _COMETRICLEARNINGI2I.fields_by_name[
228
+ 'multi_similarity_loss'].message_type = easy__rec_dot_python_dot_protos_dot_loss__pb2._MULTISIMILARITYLOSS
229
+ _COMETRICLEARNINGI2I.oneofs_by_name['loss'].fields.append(
230
+ _COMETRICLEARNINGI2I.fields_by_name['circle_loss'])
231
+ _COMETRICLEARNINGI2I.fields_by_name[
232
+ 'circle_loss'].containing_oneof = _COMETRICLEARNINGI2I.oneofs_by_name[
233
+ 'loss']
234
+ _COMETRICLEARNINGI2I.oneofs_by_name['loss'].fields.append(
235
+ _COMETRICLEARNINGI2I.fields_by_name['multi_similarity_loss'])
236
+ _COMETRICLEARNINGI2I.fields_by_name[
237
+ 'multi_similarity_loss'].containing_oneof = _COMETRICLEARNINGI2I.oneofs_by_name[
238
+ 'loss']
239
+ DESCRIPTOR.message_types_by_name['CoMetricLearningI2I'] = _COMETRICLEARNINGI2I
240
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
241
+
242
+ CoMetricLearningI2I = _reflection.GeneratedProtocolMessageType(
243
+ 'CoMetricLearningI2I',
244
+ (_message.Message,),
245
+ dict(
246
+ DESCRIPTOR=_COMETRICLEARNINGI2I,
247
+ __module__='easy_rec.python.protos.collaborative_metric_learning_pb2'
248
+ # @@protoc_insertion_point(class_scope:protos.CoMetricLearningI2I)
249
+ ))
250
+ _sym_db.RegisterMessage(CoMetricLearningI2I)
251
+
252
+ # @@protoc_insertion_point(module_scope)