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,1359 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/feature_config.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
+ from google.protobuf.internal import enum_type_wrapper
12
+
13
+ from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2 # NOQA
14
+ from easy_rec.python.protos import hyperparams_pb2 as easy__rec_dot_python_dot_protos_dot_hyperparams__pb2 # NOQA
15
+ from easy_rec.python.protos import layer_pb2 as easy__rec_dot_python_dot_protos_dot_layer__pb2 # NOQA
16
+
17
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
18
+ # @@protoc_insertion_point(imports)
19
+
20
+ _sym_db = _symbol_database.Default()
21
+
22
+ DESCRIPTOR = _descriptor.FileDescriptor(
23
+ name='easy_rec/python/protos/feature_config.proto',
24
+ package='protos',
25
+ syntax='proto2',
26
+ serialized_pb=_b(
27
+ '\n+easy_rec/python/protos/feature_config.proto\x12\x06protos\x1a(easy_rec/python/protos/hyperparams.proto\x1a easy_rec/python/protos/dnn.proto\x1a\"easy_rec/python/protos/layer.proto\"\x13\n\x11\x41ttentionCombiner\"\x1c\n\x1aMultiHeadAttentionCombiner\"\xb7\x01\n\x10SequenceCombiner\x12.\n\tattention\x18\x01 \x01(\x0b\x32\x19.protos.AttentionCombinerH\x00\x12\x42\n\x14multi_head_attention\x18\x02 \x01(\x0b\x32\".protos.MultiHeadAttentionCombinerH\x00\x12#\n\x08text_cnn\x18\x03 \x01(\x0b\x32\x0f.protos.TextCNNH\x00\x42\n\n\x08\x63ombiner\"\x96\x01\n\x08\x45VParams\x12\x16\n\x0b\x66ilter_freq\x18\x01 \x01(\x04:\x01\x30\x12\x18\n\rsteps_to_live\x18\x02 \x01(\x04:\x01\x30\x12\x18\n\tuse_cache\x18\x03 \x01(\x08:\x05\x66\x61lse\x12\x1e\n\rinit_capacity\x18\x04 \x01(\x04:\x07\x38\x33\x38\x38\x36\x30\x38\x12\x1e\n\x0cmax_capacity\x18\x05 \x01(\x04:\x08\x31\x36\x37\x37\x37\x32\x31\x36\"\xff\x08\n\rFeatureConfig\x12\x14\n\x0c\x66\x65\x61ture_name\x18\x01 \x01(\t\x12\x13\n\x0binput_names\x18\x02 \x03(\t\x12\x42\n\x0c\x66\x65\x61ture_type\x18\x03 \x02(\x0e\x32!.protos.FeatureConfig.FeatureType:\tIdFeature\x12\x18\n\x0e\x65mbedding_name\x18\x04 \x01(\t:\x00\x12\x18\n\rembedding_dim\x18\x05 \x01(\r:\x01\x30\x12\x1b\n\x10hash_bucket_size\x18\x06 \x01(\x04:\x01\x30\x12\x16\n\x0bnum_buckets\x18\x07 \x01(\x04:\x01\x30\x12\x12\n\nboundaries\x18\x08 \x03(\x01\x12\x14\n\tseparator\x18\t \x01(\t:\x01|\x12\x14\n\x0ckv_separator\x18\n \x01(\t\x12\x15\n\rseq_multi_sep\x18\x65 \x01(\t\x12\x13\n\x0bmax_seq_len\x18\x66 \x01(\r\x12\x12\n\nvocab_file\x18\x0b \x01(\t\x12\x12\n\nvocab_list\x18\x0c \x03(\t\x12\x14\n\x0cshared_names\x18\x10 \x03(\t\x12#\n\x17lookup_max_sel_elem_num\x18\x11 \x01(\x05:\x02\x31\x30\x12\x19\n\x0emax_partitions\x18\x12 \x01(\x05:\x01\x31\x12\x15\n\x08\x63ombiner\x18\x13 \x01(\t:\x03sum\x12(\n\x0binitializer\x18\x14 \x01(\x0b\x32\x13.protos.Initializer\x12\x15\n\tprecision\x18\x15 \x01(\x05:\x02-1\x12\x13\n\x07min_val\x18\xd4\x01 \x01(\x01:\x01\x30\x12\x13\n\x07max_val\x18\xd5\x01 \x01(\x01:\x01\x30\x12\x16\n\rnormalizer_fn\x18\xd6\x01 \x01(\t\x12\x18\n\rraw_input_dim\x18\x18 \x01(\r:\x01\x31\x12\x33\n\x11sequence_combiner\x18\x19 \x01(\x0b\x32\x18.protos.SequenceCombiner\x12\x46\n\x10sub_feature_type\x18\x1a \x01(\x0e\x32!.protos.FeatureConfig.FeatureType:\tIdFeature\x12\x1a\n\x0fsequence_length\x18\x1b \x01(\r:\x01\x31\x12\x12\n\nexpression\x18\x1e \x01(\t\x12#\n\tev_params\x18\x1f \x01(\x0b\x32\x10.protos.EVParams\x12\x19\n\x0e\x63ombo_join_sep\x18\x91\x03 \x01(\t:\x00\x12\x19\n\x10\x63ombo_input_seps\x18\x92\x03 \x03(\t\"\x9f\x01\n\x0b\x46\x65\x61tureType\x12\r\n\tIdFeature\x10\x00\x12\x0e\n\nRawFeature\x10\x01\x12\x0e\n\nTagFeature\x10\x02\x12\x10\n\x0c\x43omboFeature\x10\x03\x12\x11\n\rLookupFeature\x10\x04\x12\x13\n\x0fSequenceFeature\x10\x05\x12\x0f\n\x0b\x45xprFeature\x10\x06\x12\x16\n\x12PassThroughFeature\x10\x07\"N\n\tFieldType\x12\t\n\x05INT32\x10\x00\x12\t\n\x05INT64\x10\x01\x12\n\n\x06STRING\x10\x02\x12\t\n\x05\x46LOAT\x10\x04\x12\n\n\x06\x44OUBLE\x10\x05\x12\x08\n\x04\x42OOL\x10\x06\"[\n\x0f\x46\x65\x61tureConfigV2\x12\'\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32\x15.protos.FeatureConfig\x12\x1f\n\x10\x65mbedding_on_cpu\x18\x02 \x01(\x08:\x05\x66\x61lse\"\xc3\x01\n\x12\x46\x65\x61tureGroupConfig\x12\x12\n\ngroup_name\x18\x01 \x01(\t\x12\x15\n\rfeature_names\x18\x02 \x03(\t\x12+\n\twide_deep\x18\x03 \x01(\x0e\x32\x12.protos.WideOrDeep:\x04\x44\x45\x45P\x12\x34\n\x11sequence_features\x18\x04 \x03(\x0b\x32\x19.protos.SeqAttGroupConfig\x12\x1f\n\x10negative_sampler\x18\x05 \x01(\x08:\x05\x66\x61lse\"@\n\tSeqAttMap\x12\x0b\n\x03key\x18\x01 \x03(\t\x12\x10\n\x08hist_seq\x18\x02 \x03(\t\x12\x14\n\x0c\x61ux_hist_seq\x18\x03 \x03(\t\"\x8b\x02\n\x11SeqAttGroupConfig\x12\x12\n\ngroup_name\x18\x01 \x01(\t\x12&\n\x0bseq_att_map\x18\x02 \x03(\x0b\x32\x11.protos.SeqAttMap\x12\x19\n\ntf_summary\x18\x03 \x01(\x08:\x05\x66\x61lse\x12\x1c\n\x07seq_dnn\x18\x04 \x01(\x0b\x32\x0b.protos.DNN\x12\x1f\n\x10\x61llow_key_search\x18\x05 \x01(\x08:\x05\x66\x61lse\x12\x1e\n\x10need_key_feature\x18\x06 \x01(\x08:\x04true\x12\"\n\x13\x61llow_key_transform\x18\x07 \x01(\x08:\x05\x66\x61lse\x12\x1c\n\rtransform_dnn\x18\x08 \x01(\x08:\x05\x66\x61lse*3\n\nWideOrDeep\x12\x08\n\x04\x44\x45\x45P\x10\x00\x12\x08\n\x04WIDE\x10\x01\x12\x11\n\rWIDE_AND_DEEP\x10\x02'
28
+ ),
29
+ dependencies=[
30
+ easy__rec_dot_python_dot_protos_dot_hyperparams__pb2.DESCRIPTOR,
31
+ easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
32
+ easy__rec_dot_python_dot_protos_dot_layer__pb2.DESCRIPTOR,
33
+ ])
34
+
35
+ _WIDEORDEEP = _descriptor.EnumDescriptor(
36
+ name='WideOrDeep',
37
+ full_name='protos.WideOrDeep',
38
+ filename=None,
39
+ file=DESCRIPTOR,
40
+ values=[
41
+ _descriptor.EnumValueDescriptor(
42
+ name='DEEP', index=0, number=0, options=None, type=None),
43
+ _descriptor.EnumValueDescriptor(
44
+ name='WIDE', index=1, number=1, options=None, type=None),
45
+ _descriptor.EnumValueDescriptor(
46
+ name='WIDE_AND_DEEP', index=2, number=2, options=None, type=None),
47
+ ],
48
+ containing_type=None,
49
+ options=None,
50
+ serialized_start=2338,
51
+ serialized_end=2389,
52
+ )
53
+ _sym_db.RegisterEnumDescriptor(_WIDEORDEEP)
54
+
55
+ WideOrDeep = enum_type_wrapper.EnumTypeWrapper(_WIDEORDEEP)
56
+ DEEP = 0
57
+ WIDE = 1
58
+ WIDE_AND_DEEP = 2
59
+
60
+ _FEATURECONFIG_FEATURETYPE = _descriptor.EnumDescriptor(
61
+ name='FeatureType',
62
+ full_name='protos.FeatureConfig.FeatureType',
63
+ filename=None,
64
+ file=DESCRIPTOR,
65
+ values=[
66
+ _descriptor.EnumValueDescriptor(
67
+ name='IdFeature', index=0, number=0, options=None, type=None),
68
+ _descriptor.EnumValueDescriptor(
69
+ name='RawFeature', index=1, number=1, options=None, type=None),
70
+ _descriptor.EnumValueDescriptor(
71
+ name='TagFeature', index=2, number=2, options=None, type=None),
72
+ _descriptor.EnumValueDescriptor(
73
+ name='ComboFeature', index=3, number=3, options=None, type=None),
74
+ _descriptor.EnumValueDescriptor(
75
+ name='LookupFeature', index=4, number=4, options=None, type=None),
76
+ _descriptor.EnumValueDescriptor(
77
+ name='SequenceFeature', index=5, number=5, options=None, type=None),
78
+ _descriptor.EnumValueDescriptor(
79
+ name='ExprFeature', index=6, number=6, options=None, type=None),
80
+ _descriptor.EnumValueDescriptor(
81
+ name='PassThroughFeature',
82
+ index=7,
83
+ number=7,
84
+ options=None,
85
+ type=None),
86
+ ],
87
+ containing_type=None,
88
+ options=None,
89
+ serialized_start=1470,
90
+ serialized_end=1629,
91
+ )
92
+ _sym_db.RegisterEnumDescriptor(_FEATURECONFIG_FEATURETYPE)
93
+
94
+ _FEATURECONFIG_FIELDTYPE = _descriptor.EnumDescriptor(
95
+ name='FieldType',
96
+ full_name='protos.FeatureConfig.FieldType',
97
+ filename=None,
98
+ file=DESCRIPTOR,
99
+ values=[
100
+ _descriptor.EnumValueDescriptor(
101
+ name='INT32', index=0, number=0, options=None, type=None),
102
+ _descriptor.EnumValueDescriptor(
103
+ name='INT64', index=1, number=1, options=None, type=None),
104
+ _descriptor.EnumValueDescriptor(
105
+ name='STRING', index=2, number=2, options=None, type=None),
106
+ _descriptor.EnumValueDescriptor(
107
+ name='FLOAT', index=3, number=4, options=None, type=None),
108
+ _descriptor.EnumValueDescriptor(
109
+ name='DOUBLE', index=4, number=5, options=None, type=None),
110
+ _descriptor.EnumValueDescriptor(
111
+ name='BOOL', index=5, number=6, options=None, type=None),
112
+ ],
113
+ containing_type=None,
114
+ options=None,
115
+ serialized_start=1631,
116
+ serialized_end=1709,
117
+ )
118
+ _sym_db.RegisterEnumDescriptor(_FEATURECONFIG_FIELDTYPE)
119
+
120
+ _ATTENTIONCOMBINER = _descriptor.Descriptor(
121
+ name='AttentionCombiner',
122
+ full_name='protos.AttentionCombiner',
123
+ filename=None,
124
+ file=DESCRIPTOR,
125
+ containing_type=None,
126
+ fields=[],
127
+ extensions=[],
128
+ nested_types=[],
129
+ enum_types=[],
130
+ options=None,
131
+ is_extendable=False,
132
+ syntax='proto2',
133
+ extension_ranges=[],
134
+ oneofs=[],
135
+ serialized_start=167,
136
+ serialized_end=186,
137
+ )
138
+
139
+ _MULTIHEADATTENTIONCOMBINER = _descriptor.Descriptor(
140
+ name='MultiHeadAttentionCombiner',
141
+ full_name='protos.MultiHeadAttentionCombiner',
142
+ filename=None,
143
+ file=DESCRIPTOR,
144
+ containing_type=None,
145
+ fields=[],
146
+ extensions=[],
147
+ nested_types=[],
148
+ enum_types=[],
149
+ options=None,
150
+ is_extendable=False,
151
+ syntax='proto2',
152
+ extension_ranges=[],
153
+ oneofs=[],
154
+ serialized_start=188,
155
+ serialized_end=216,
156
+ )
157
+
158
+ _SEQUENCECOMBINER = _descriptor.Descriptor(
159
+ name='SequenceCombiner',
160
+ full_name='protos.SequenceCombiner',
161
+ filename=None,
162
+ file=DESCRIPTOR,
163
+ containing_type=None,
164
+ fields=[
165
+ _descriptor.FieldDescriptor(
166
+ name='attention',
167
+ full_name='protos.SequenceCombiner.attention',
168
+ index=0,
169
+ number=1,
170
+ type=11,
171
+ cpp_type=10,
172
+ label=1,
173
+ has_default_value=False,
174
+ default_value=None,
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='multi_head_attention',
183
+ full_name='protos.SequenceCombiner.multi_head_attention',
184
+ index=1,
185
+ number=2,
186
+ type=11,
187
+ cpp_type=10,
188
+ label=1,
189
+ has_default_value=False,
190
+ default_value=None,
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='text_cnn',
199
+ full_name='protos.SequenceCombiner.text_cnn',
200
+ index=2,
201
+ number=3,
202
+ type=11,
203
+ cpp_type=10,
204
+ label=1,
205
+ has_default_value=False,
206
+ default_value=None,
207
+ message_type=None,
208
+ enum_type=None,
209
+ containing_type=None,
210
+ is_extension=False,
211
+ extension_scope=None,
212
+ options=None),
213
+ ],
214
+ extensions=[],
215
+ nested_types=[],
216
+ enum_types=[],
217
+ options=None,
218
+ is_extendable=False,
219
+ syntax='proto2',
220
+ extension_ranges=[],
221
+ oneofs=[
222
+ _descriptor.OneofDescriptor(
223
+ name='combiner',
224
+ full_name='protos.SequenceCombiner.combiner',
225
+ index=0,
226
+ containing_type=None,
227
+ fields=[]),
228
+ ],
229
+ serialized_start=219,
230
+ serialized_end=402,
231
+ )
232
+
233
+ _EVPARAMS = _descriptor.Descriptor(
234
+ name='EVParams',
235
+ full_name='protos.EVParams',
236
+ filename=None,
237
+ file=DESCRIPTOR,
238
+ containing_type=None,
239
+ fields=[
240
+ _descriptor.FieldDescriptor(
241
+ name='filter_freq',
242
+ full_name='protos.EVParams.filter_freq',
243
+ index=0,
244
+ number=1,
245
+ type=4,
246
+ cpp_type=4,
247
+ label=1,
248
+ has_default_value=True,
249
+ default_value=0,
250
+ message_type=None,
251
+ enum_type=None,
252
+ containing_type=None,
253
+ is_extension=False,
254
+ extension_scope=None,
255
+ options=None),
256
+ _descriptor.FieldDescriptor(
257
+ name='steps_to_live',
258
+ full_name='protos.EVParams.steps_to_live',
259
+ index=1,
260
+ number=2,
261
+ type=4,
262
+ cpp_type=4,
263
+ label=1,
264
+ has_default_value=True,
265
+ default_value=0,
266
+ message_type=None,
267
+ enum_type=None,
268
+ containing_type=None,
269
+ is_extension=False,
270
+ extension_scope=None,
271
+ options=None),
272
+ _descriptor.FieldDescriptor(
273
+ name='use_cache',
274
+ full_name='protos.EVParams.use_cache',
275
+ index=2,
276
+ number=3,
277
+ type=8,
278
+ cpp_type=7,
279
+ label=1,
280
+ has_default_value=True,
281
+ default_value=False,
282
+ message_type=None,
283
+ enum_type=None,
284
+ containing_type=None,
285
+ is_extension=False,
286
+ extension_scope=None,
287
+ options=None),
288
+ _descriptor.FieldDescriptor(
289
+ name='init_capacity',
290
+ full_name='protos.EVParams.init_capacity',
291
+ index=3,
292
+ number=4,
293
+ type=4,
294
+ cpp_type=4,
295
+ label=1,
296
+ has_default_value=True,
297
+ default_value=8388608,
298
+ message_type=None,
299
+ enum_type=None,
300
+ containing_type=None,
301
+ is_extension=False,
302
+ extension_scope=None,
303
+ options=None),
304
+ _descriptor.FieldDescriptor(
305
+ name='max_capacity',
306
+ full_name='protos.EVParams.max_capacity',
307
+ index=4,
308
+ number=5,
309
+ type=4,
310
+ cpp_type=4,
311
+ label=1,
312
+ has_default_value=True,
313
+ default_value=16777216,
314
+ message_type=None,
315
+ enum_type=None,
316
+ containing_type=None,
317
+ is_extension=False,
318
+ extension_scope=None,
319
+ options=None),
320
+ ],
321
+ extensions=[],
322
+ nested_types=[],
323
+ enum_types=[],
324
+ options=None,
325
+ is_extendable=False,
326
+ syntax='proto2',
327
+ extension_ranges=[],
328
+ oneofs=[],
329
+ serialized_start=405,
330
+ serialized_end=555,
331
+ )
332
+
333
+ _FEATURECONFIG = _descriptor.Descriptor(
334
+ name='FeatureConfig',
335
+ full_name='protos.FeatureConfig',
336
+ filename=None,
337
+ file=DESCRIPTOR,
338
+ containing_type=None,
339
+ fields=[
340
+ _descriptor.FieldDescriptor(
341
+ name='feature_name',
342
+ full_name='protos.FeatureConfig.feature_name',
343
+ index=0,
344
+ number=1,
345
+ type=9,
346
+ cpp_type=9,
347
+ label=1,
348
+ has_default_value=False,
349
+ default_value=_b('').decode('utf-8'),
350
+ message_type=None,
351
+ enum_type=None,
352
+ containing_type=None,
353
+ is_extension=False,
354
+ extension_scope=None,
355
+ options=None),
356
+ _descriptor.FieldDescriptor(
357
+ name='input_names',
358
+ full_name='protos.FeatureConfig.input_names',
359
+ index=1,
360
+ number=2,
361
+ type=9,
362
+ cpp_type=9,
363
+ label=3,
364
+ has_default_value=False,
365
+ default_value=[],
366
+ message_type=None,
367
+ enum_type=None,
368
+ containing_type=None,
369
+ is_extension=False,
370
+ extension_scope=None,
371
+ options=None),
372
+ _descriptor.FieldDescriptor(
373
+ name='feature_type',
374
+ full_name='protos.FeatureConfig.feature_type',
375
+ index=2,
376
+ number=3,
377
+ type=14,
378
+ cpp_type=8,
379
+ label=2,
380
+ has_default_value=True,
381
+ default_value=0,
382
+ message_type=None,
383
+ enum_type=None,
384
+ containing_type=None,
385
+ is_extension=False,
386
+ extension_scope=None,
387
+ options=None),
388
+ _descriptor.FieldDescriptor(
389
+ name='embedding_name',
390
+ full_name='protos.FeatureConfig.embedding_name',
391
+ index=3,
392
+ number=4,
393
+ type=9,
394
+ cpp_type=9,
395
+ label=1,
396
+ has_default_value=True,
397
+ default_value=_b('').decode('utf-8'),
398
+ message_type=None,
399
+ enum_type=None,
400
+ containing_type=None,
401
+ is_extension=False,
402
+ extension_scope=None,
403
+ options=None),
404
+ _descriptor.FieldDescriptor(
405
+ name='embedding_dim',
406
+ full_name='protos.FeatureConfig.embedding_dim',
407
+ index=4,
408
+ number=5,
409
+ type=13,
410
+ cpp_type=3,
411
+ label=1,
412
+ has_default_value=True,
413
+ default_value=0,
414
+ message_type=None,
415
+ enum_type=None,
416
+ containing_type=None,
417
+ is_extension=False,
418
+ extension_scope=None,
419
+ options=None),
420
+ _descriptor.FieldDescriptor(
421
+ name='hash_bucket_size',
422
+ full_name='protos.FeatureConfig.hash_bucket_size',
423
+ index=5,
424
+ number=6,
425
+ type=4,
426
+ cpp_type=4,
427
+ label=1,
428
+ has_default_value=True,
429
+ default_value=0,
430
+ message_type=None,
431
+ enum_type=None,
432
+ containing_type=None,
433
+ is_extension=False,
434
+ extension_scope=None,
435
+ options=None),
436
+ _descriptor.FieldDescriptor(
437
+ name='num_buckets',
438
+ full_name='protos.FeatureConfig.num_buckets',
439
+ index=6,
440
+ number=7,
441
+ type=4,
442
+ cpp_type=4,
443
+ label=1,
444
+ has_default_value=True,
445
+ default_value=0,
446
+ message_type=None,
447
+ enum_type=None,
448
+ containing_type=None,
449
+ is_extension=False,
450
+ extension_scope=None,
451
+ options=None),
452
+ _descriptor.FieldDescriptor(
453
+ name='boundaries',
454
+ full_name='protos.FeatureConfig.boundaries',
455
+ index=7,
456
+ number=8,
457
+ type=1,
458
+ cpp_type=5,
459
+ label=3,
460
+ has_default_value=False,
461
+ default_value=[],
462
+ message_type=None,
463
+ enum_type=None,
464
+ containing_type=None,
465
+ is_extension=False,
466
+ extension_scope=None,
467
+ options=None),
468
+ _descriptor.FieldDescriptor(
469
+ name='separator',
470
+ full_name='protos.FeatureConfig.separator',
471
+ index=8,
472
+ number=9,
473
+ type=9,
474
+ cpp_type=9,
475
+ label=1,
476
+ has_default_value=True,
477
+ default_value=_b('|').decode('utf-8'),
478
+ message_type=None,
479
+ enum_type=None,
480
+ containing_type=None,
481
+ is_extension=False,
482
+ extension_scope=None,
483
+ options=None),
484
+ _descriptor.FieldDescriptor(
485
+ name='kv_separator',
486
+ full_name='protos.FeatureConfig.kv_separator',
487
+ index=9,
488
+ number=10,
489
+ type=9,
490
+ cpp_type=9,
491
+ label=1,
492
+ has_default_value=False,
493
+ default_value=_b('').decode('utf-8'),
494
+ message_type=None,
495
+ enum_type=None,
496
+ containing_type=None,
497
+ is_extension=False,
498
+ extension_scope=None,
499
+ options=None),
500
+ _descriptor.FieldDescriptor(
501
+ name='seq_multi_sep',
502
+ full_name='protos.FeatureConfig.seq_multi_sep',
503
+ index=10,
504
+ number=101,
505
+ type=9,
506
+ cpp_type=9,
507
+ label=1,
508
+ has_default_value=False,
509
+ default_value=_b('').decode('utf-8'),
510
+ message_type=None,
511
+ enum_type=None,
512
+ containing_type=None,
513
+ is_extension=False,
514
+ extension_scope=None,
515
+ options=None),
516
+ _descriptor.FieldDescriptor(
517
+ name='max_seq_len',
518
+ full_name='protos.FeatureConfig.max_seq_len',
519
+ index=11,
520
+ number=102,
521
+ type=13,
522
+ cpp_type=3,
523
+ label=1,
524
+ has_default_value=False,
525
+ default_value=0,
526
+ message_type=None,
527
+ enum_type=None,
528
+ containing_type=None,
529
+ is_extension=False,
530
+ extension_scope=None,
531
+ options=None),
532
+ _descriptor.FieldDescriptor(
533
+ name='vocab_file',
534
+ full_name='protos.FeatureConfig.vocab_file',
535
+ index=12,
536
+ number=11,
537
+ type=9,
538
+ cpp_type=9,
539
+ label=1,
540
+ has_default_value=False,
541
+ default_value=_b('').decode('utf-8'),
542
+ message_type=None,
543
+ enum_type=None,
544
+ containing_type=None,
545
+ is_extension=False,
546
+ extension_scope=None,
547
+ options=None),
548
+ _descriptor.FieldDescriptor(
549
+ name='vocab_list',
550
+ full_name='protos.FeatureConfig.vocab_list',
551
+ index=13,
552
+ number=12,
553
+ type=9,
554
+ cpp_type=9,
555
+ label=3,
556
+ has_default_value=False,
557
+ default_value=[],
558
+ message_type=None,
559
+ enum_type=None,
560
+ containing_type=None,
561
+ is_extension=False,
562
+ extension_scope=None,
563
+ options=None),
564
+ _descriptor.FieldDescriptor(
565
+ name='shared_names',
566
+ full_name='protos.FeatureConfig.shared_names',
567
+ index=14,
568
+ number=16,
569
+ type=9,
570
+ cpp_type=9,
571
+ label=3,
572
+ has_default_value=False,
573
+ default_value=[],
574
+ message_type=None,
575
+ enum_type=None,
576
+ containing_type=None,
577
+ is_extension=False,
578
+ extension_scope=None,
579
+ options=None),
580
+ _descriptor.FieldDescriptor(
581
+ name='lookup_max_sel_elem_num',
582
+ full_name='protos.FeatureConfig.lookup_max_sel_elem_num',
583
+ index=15,
584
+ number=17,
585
+ type=5,
586
+ cpp_type=1,
587
+ label=1,
588
+ has_default_value=True,
589
+ default_value=10,
590
+ message_type=None,
591
+ enum_type=None,
592
+ containing_type=None,
593
+ is_extension=False,
594
+ extension_scope=None,
595
+ options=None),
596
+ _descriptor.FieldDescriptor(
597
+ name='max_partitions',
598
+ full_name='protos.FeatureConfig.max_partitions',
599
+ index=16,
600
+ number=18,
601
+ type=5,
602
+ cpp_type=1,
603
+ label=1,
604
+ has_default_value=True,
605
+ default_value=1,
606
+ message_type=None,
607
+ enum_type=None,
608
+ containing_type=None,
609
+ is_extension=False,
610
+ extension_scope=None,
611
+ options=None),
612
+ _descriptor.FieldDescriptor(
613
+ name='combiner',
614
+ full_name='protos.FeatureConfig.combiner',
615
+ index=17,
616
+ number=19,
617
+ type=9,
618
+ cpp_type=9,
619
+ label=1,
620
+ has_default_value=True,
621
+ default_value=_b('sum').decode('utf-8'),
622
+ message_type=None,
623
+ enum_type=None,
624
+ containing_type=None,
625
+ is_extension=False,
626
+ extension_scope=None,
627
+ options=None),
628
+ _descriptor.FieldDescriptor(
629
+ name='initializer',
630
+ full_name='protos.FeatureConfig.initializer',
631
+ index=18,
632
+ number=20,
633
+ type=11,
634
+ cpp_type=10,
635
+ label=1,
636
+ has_default_value=False,
637
+ default_value=None,
638
+ message_type=None,
639
+ enum_type=None,
640
+ containing_type=None,
641
+ is_extension=False,
642
+ extension_scope=None,
643
+ options=None),
644
+ _descriptor.FieldDescriptor(
645
+ name='precision',
646
+ full_name='protos.FeatureConfig.precision',
647
+ index=19,
648
+ number=21,
649
+ type=5,
650
+ cpp_type=1,
651
+ label=1,
652
+ has_default_value=True,
653
+ default_value=-1,
654
+ message_type=None,
655
+ enum_type=None,
656
+ containing_type=None,
657
+ is_extension=False,
658
+ extension_scope=None,
659
+ options=None),
660
+ _descriptor.FieldDescriptor(
661
+ name='min_val',
662
+ full_name='protos.FeatureConfig.min_val',
663
+ index=20,
664
+ number=212,
665
+ type=1,
666
+ cpp_type=5,
667
+ label=1,
668
+ has_default_value=True,
669
+ default_value=float(0),
670
+ message_type=None,
671
+ enum_type=None,
672
+ containing_type=None,
673
+ is_extension=False,
674
+ extension_scope=None,
675
+ options=None),
676
+ _descriptor.FieldDescriptor(
677
+ name='max_val',
678
+ full_name='protos.FeatureConfig.max_val',
679
+ index=21,
680
+ number=213,
681
+ type=1,
682
+ cpp_type=5,
683
+ label=1,
684
+ has_default_value=True,
685
+ default_value=float(0),
686
+ message_type=None,
687
+ enum_type=None,
688
+ containing_type=None,
689
+ is_extension=False,
690
+ extension_scope=None,
691
+ options=None),
692
+ _descriptor.FieldDescriptor(
693
+ name='normalizer_fn',
694
+ full_name='protos.FeatureConfig.normalizer_fn',
695
+ index=22,
696
+ number=214,
697
+ type=9,
698
+ cpp_type=9,
699
+ label=1,
700
+ has_default_value=False,
701
+ default_value=_b('').decode('utf-8'),
702
+ message_type=None,
703
+ enum_type=None,
704
+ containing_type=None,
705
+ is_extension=False,
706
+ extension_scope=None,
707
+ options=None),
708
+ _descriptor.FieldDescriptor(
709
+ name='raw_input_dim',
710
+ full_name='protos.FeatureConfig.raw_input_dim',
711
+ index=23,
712
+ number=24,
713
+ type=13,
714
+ cpp_type=3,
715
+ label=1,
716
+ has_default_value=True,
717
+ default_value=1,
718
+ message_type=None,
719
+ enum_type=None,
720
+ containing_type=None,
721
+ is_extension=False,
722
+ extension_scope=None,
723
+ options=None),
724
+ _descriptor.FieldDescriptor(
725
+ name='sequence_combiner',
726
+ full_name='protos.FeatureConfig.sequence_combiner',
727
+ index=24,
728
+ number=25,
729
+ type=11,
730
+ cpp_type=10,
731
+ label=1,
732
+ has_default_value=False,
733
+ default_value=None,
734
+ message_type=None,
735
+ enum_type=None,
736
+ containing_type=None,
737
+ is_extension=False,
738
+ extension_scope=None,
739
+ options=None),
740
+ _descriptor.FieldDescriptor(
741
+ name='sub_feature_type',
742
+ full_name='protos.FeatureConfig.sub_feature_type',
743
+ index=25,
744
+ number=26,
745
+ type=14,
746
+ cpp_type=8,
747
+ label=1,
748
+ has_default_value=True,
749
+ default_value=0,
750
+ message_type=None,
751
+ enum_type=None,
752
+ containing_type=None,
753
+ is_extension=False,
754
+ extension_scope=None,
755
+ options=None),
756
+ _descriptor.FieldDescriptor(
757
+ name='sequence_length',
758
+ full_name='protos.FeatureConfig.sequence_length',
759
+ index=26,
760
+ number=27,
761
+ type=13,
762
+ cpp_type=3,
763
+ label=1,
764
+ has_default_value=True,
765
+ default_value=1,
766
+ message_type=None,
767
+ enum_type=None,
768
+ containing_type=None,
769
+ is_extension=False,
770
+ extension_scope=None,
771
+ options=None),
772
+ _descriptor.FieldDescriptor(
773
+ name='expression',
774
+ full_name='protos.FeatureConfig.expression',
775
+ index=27,
776
+ number=30,
777
+ type=9,
778
+ cpp_type=9,
779
+ label=1,
780
+ has_default_value=False,
781
+ default_value=_b('').decode('utf-8'),
782
+ message_type=None,
783
+ enum_type=None,
784
+ containing_type=None,
785
+ is_extension=False,
786
+ extension_scope=None,
787
+ options=None),
788
+ _descriptor.FieldDescriptor(
789
+ name='ev_params',
790
+ full_name='protos.FeatureConfig.ev_params',
791
+ index=28,
792
+ number=31,
793
+ type=11,
794
+ cpp_type=10,
795
+ label=1,
796
+ has_default_value=False,
797
+ default_value=None,
798
+ message_type=None,
799
+ enum_type=None,
800
+ containing_type=None,
801
+ is_extension=False,
802
+ extension_scope=None,
803
+ options=None),
804
+ _descriptor.FieldDescriptor(
805
+ name='combo_join_sep',
806
+ full_name='protos.FeatureConfig.combo_join_sep',
807
+ index=29,
808
+ number=401,
809
+ type=9,
810
+ cpp_type=9,
811
+ label=1,
812
+ has_default_value=True,
813
+ default_value=_b('').decode('utf-8'),
814
+ message_type=None,
815
+ enum_type=None,
816
+ containing_type=None,
817
+ is_extension=False,
818
+ extension_scope=None,
819
+ options=None),
820
+ _descriptor.FieldDescriptor(
821
+ name='combo_input_seps',
822
+ full_name='protos.FeatureConfig.combo_input_seps',
823
+ index=30,
824
+ number=402,
825
+ type=9,
826
+ cpp_type=9,
827
+ label=3,
828
+ has_default_value=False,
829
+ default_value=[],
830
+ message_type=None,
831
+ enum_type=None,
832
+ containing_type=None,
833
+ is_extension=False,
834
+ extension_scope=None,
835
+ options=None),
836
+ ],
837
+ extensions=[],
838
+ nested_types=[],
839
+ enum_types=[
840
+ _FEATURECONFIG_FEATURETYPE,
841
+ _FEATURECONFIG_FIELDTYPE,
842
+ ],
843
+ options=None,
844
+ is_extendable=False,
845
+ syntax='proto2',
846
+ extension_ranges=[],
847
+ oneofs=[],
848
+ serialized_start=558,
849
+ serialized_end=1709,
850
+ )
851
+
852
+ _FEATURECONFIGV2 = _descriptor.Descriptor(
853
+ name='FeatureConfigV2',
854
+ full_name='protos.FeatureConfigV2',
855
+ filename=None,
856
+ file=DESCRIPTOR,
857
+ containing_type=None,
858
+ fields=[
859
+ _descriptor.FieldDescriptor(
860
+ name='features',
861
+ full_name='protos.FeatureConfigV2.features',
862
+ index=0,
863
+ number=1,
864
+ type=11,
865
+ cpp_type=10,
866
+ label=3,
867
+ has_default_value=False,
868
+ default_value=[],
869
+ message_type=None,
870
+ enum_type=None,
871
+ containing_type=None,
872
+ is_extension=False,
873
+ extension_scope=None,
874
+ options=None),
875
+ _descriptor.FieldDescriptor(
876
+ name='embedding_on_cpu',
877
+ full_name='protos.FeatureConfigV2.embedding_on_cpu',
878
+ index=1,
879
+ number=2,
880
+ type=8,
881
+ cpp_type=7,
882
+ label=1,
883
+ has_default_value=True,
884
+ default_value=False,
885
+ message_type=None,
886
+ enum_type=None,
887
+ containing_type=None,
888
+ is_extension=False,
889
+ extension_scope=None,
890
+ options=None),
891
+ ],
892
+ extensions=[],
893
+ nested_types=[],
894
+ enum_types=[],
895
+ options=None,
896
+ is_extendable=False,
897
+ syntax='proto2',
898
+ extension_ranges=[],
899
+ oneofs=[],
900
+ serialized_start=1711,
901
+ serialized_end=1802,
902
+ )
903
+
904
+ _FEATUREGROUPCONFIG = _descriptor.Descriptor(
905
+ name='FeatureGroupConfig',
906
+ full_name='protos.FeatureGroupConfig',
907
+ filename=None,
908
+ file=DESCRIPTOR,
909
+ containing_type=None,
910
+ fields=[
911
+ _descriptor.FieldDescriptor(
912
+ name='group_name',
913
+ full_name='protos.FeatureGroupConfig.group_name',
914
+ index=0,
915
+ number=1,
916
+ type=9,
917
+ cpp_type=9,
918
+ label=1,
919
+ has_default_value=False,
920
+ default_value=_b('').decode('utf-8'),
921
+ message_type=None,
922
+ enum_type=None,
923
+ containing_type=None,
924
+ is_extension=False,
925
+ extension_scope=None,
926
+ options=None),
927
+ _descriptor.FieldDescriptor(
928
+ name='feature_names',
929
+ full_name='protos.FeatureGroupConfig.feature_names',
930
+ index=1,
931
+ number=2,
932
+ type=9,
933
+ cpp_type=9,
934
+ label=3,
935
+ has_default_value=False,
936
+ default_value=[],
937
+ message_type=None,
938
+ enum_type=None,
939
+ containing_type=None,
940
+ is_extension=False,
941
+ extension_scope=None,
942
+ options=None),
943
+ _descriptor.FieldDescriptor(
944
+ name='wide_deep',
945
+ full_name='protos.FeatureGroupConfig.wide_deep',
946
+ index=2,
947
+ number=3,
948
+ type=14,
949
+ cpp_type=8,
950
+ label=1,
951
+ has_default_value=True,
952
+ default_value=0,
953
+ message_type=None,
954
+ enum_type=None,
955
+ containing_type=None,
956
+ is_extension=False,
957
+ extension_scope=None,
958
+ options=None),
959
+ _descriptor.FieldDescriptor(
960
+ name='sequence_features',
961
+ full_name='protos.FeatureGroupConfig.sequence_features',
962
+ index=3,
963
+ number=4,
964
+ type=11,
965
+ cpp_type=10,
966
+ label=3,
967
+ has_default_value=False,
968
+ default_value=[],
969
+ message_type=None,
970
+ enum_type=None,
971
+ containing_type=None,
972
+ is_extension=False,
973
+ extension_scope=None,
974
+ options=None),
975
+ _descriptor.FieldDescriptor(
976
+ name='negative_sampler',
977
+ full_name='protos.FeatureGroupConfig.negative_sampler',
978
+ index=4,
979
+ number=5,
980
+ type=8,
981
+ cpp_type=7,
982
+ label=1,
983
+ has_default_value=True,
984
+ default_value=False,
985
+ message_type=None,
986
+ enum_type=None,
987
+ containing_type=None,
988
+ is_extension=False,
989
+ extension_scope=None,
990
+ options=None),
991
+ ],
992
+ extensions=[],
993
+ nested_types=[],
994
+ enum_types=[],
995
+ options=None,
996
+ is_extendable=False,
997
+ syntax='proto2',
998
+ extension_ranges=[],
999
+ oneofs=[],
1000
+ serialized_start=1805,
1001
+ serialized_end=2000,
1002
+ )
1003
+
1004
+ _SEQATTMAP = _descriptor.Descriptor(
1005
+ name='SeqAttMap',
1006
+ full_name='protos.SeqAttMap',
1007
+ filename=None,
1008
+ file=DESCRIPTOR,
1009
+ containing_type=None,
1010
+ fields=[
1011
+ _descriptor.FieldDescriptor(
1012
+ name='key',
1013
+ full_name='protos.SeqAttMap.key',
1014
+ index=0,
1015
+ number=1,
1016
+ type=9,
1017
+ cpp_type=9,
1018
+ label=3,
1019
+ has_default_value=False,
1020
+ default_value=[],
1021
+ message_type=None,
1022
+ enum_type=None,
1023
+ containing_type=None,
1024
+ is_extension=False,
1025
+ extension_scope=None,
1026
+ options=None),
1027
+ _descriptor.FieldDescriptor(
1028
+ name='hist_seq',
1029
+ full_name='protos.SeqAttMap.hist_seq',
1030
+ index=1,
1031
+ number=2,
1032
+ type=9,
1033
+ cpp_type=9,
1034
+ label=3,
1035
+ has_default_value=False,
1036
+ default_value=[],
1037
+ message_type=None,
1038
+ enum_type=None,
1039
+ containing_type=None,
1040
+ is_extension=False,
1041
+ extension_scope=None,
1042
+ options=None),
1043
+ _descriptor.FieldDescriptor(
1044
+ name='aux_hist_seq',
1045
+ full_name='protos.SeqAttMap.aux_hist_seq',
1046
+ index=2,
1047
+ number=3,
1048
+ type=9,
1049
+ cpp_type=9,
1050
+ label=3,
1051
+ has_default_value=False,
1052
+ default_value=[],
1053
+ message_type=None,
1054
+ enum_type=None,
1055
+ containing_type=None,
1056
+ is_extension=False,
1057
+ extension_scope=None,
1058
+ options=None),
1059
+ ],
1060
+ extensions=[],
1061
+ nested_types=[],
1062
+ enum_types=[],
1063
+ options=None,
1064
+ is_extendable=False,
1065
+ syntax='proto2',
1066
+ extension_ranges=[],
1067
+ oneofs=[],
1068
+ serialized_start=2002,
1069
+ serialized_end=2066,
1070
+ )
1071
+
1072
+ _SEQATTGROUPCONFIG = _descriptor.Descriptor(
1073
+ name='SeqAttGroupConfig',
1074
+ full_name='protos.SeqAttGroupConfig',
1075
+ filename=None,
1076
+ file=DESCRIPTOR,
1077
+ containing_type=None,
1078
+ fields=[
1079
+ _descriptor.FieldDescriptor(
1080
+ name='group_name',
1081
+ full_name='protos.SeqAttGroupConfig.group_name',
1082
+ index=0,
1083
+ number=1,
1084
+ type=9,
1085
+ cpp_type=9,
1086
+ label=1,
1087
+ has_default_value=False,
1088
+ default_value=_b('').decode('utf-8'),
1089
+ message_type=None,
1090
+ enum_type=None,
1091
+ containing_type=None,
1092
+ is_extension=False,
1093
+ extension_scope=None,
1094
+ options=None),
1095
+ _descriptor.FieldDescriptor(
1096
+ name='seq_att_map',
1097
+ full_name='protos.SeqAttGroupConfig.seq_att_map',
1098
+ index=1,
1099
+ number=2,
1100
+ type=11,
1101
+ cpp_type=10,
1102
+ label=3,
1103
+ has_default_value=False,
1104
+ default_value=[],
1105
+ message_type=None,
1106
+ enum_type=None,
1107
+ containing_type=None,
1108
+ is_extension=False,
1109
+ extension_scope=None,
1110
+ options=None),
1111
+ _descriptor.FieldDescriptor(
1112
+ name='tf_summary',
1113
+ full_name='protos.SeqAttGroupConfig.tf_summary',
1114
+ index=2,
1115
+ number=3,
1116
+ type=8,
1117
+ cpp_type=7,
1118
+ label=1,
1119
+ has_default_value=True,
1120
+ default_value=False,
1121
+ message_type=None,
1122
+ enum_type=None,
1123
+ containing_type=None,
1124
+ is_extension=False,
1125
+ extension_scope=None,
1126
+ options=None),
1127
+ _descriptor.FieldDescriptor(
1128
+ name='seq_dnn',
1129
+ full_name='protos.SeqAttGroupConfig.seq_dnn',
1130
+ index=3,
1131
+ number=4,
1132
+ type=11,
1133
+ cpp_type=10,
1134
+ label=1,
1135
+ has_default_value=False,
1136
+ default_value=None,
1137
+ message_type=None,
1138
+ enum_type=None,
1139
+ containing_type=None,
1140
+ is_extension=False,
1141
+ extension_scope=None,
1142
+ options=None),
1143
+ _descriptor.FieldDescriptor(
1144
+ name='allow_key_search',
1145
+ full_name='protos.SeqAttGroupConfig.allow_key_search',
1146
+ index=4,
1147
+ number=5,
1148
+ type=8,
1149
+ cpp_type=7,
1150
+ label=1,
1151
+ has_default_value=True,
1152
+ default_value=False,
1153
+ message_type=None,
1154
+ enum_type=None,
1155
+ containing_type=None,
1156
+ is_extension=False,
1157
+ extension_scope=None,
1158
+ options=None),
1159
+ _descriptor.FieldDescriptor(
1160
+ name='need_key_feature',
1161
+ full_name='protos.SeqAttGroupConfig.need_key_feature',
1162
+ index=5,
1163
+ number=6,
1164
+ type=8,
1165
+ cpp_type=7,
1166
+ label=1,
1167
+ has_default_value=True,
1168
+ default_value=True,
1169
+ message_type=None,
1170
+ enum_type=None,
1171
+ containing_type=None,
1172
+ is_extension=False,
1173
+ extension_scope=None,
1174
+ options=None),
1175
+ _descriptor.FieldDescriptor(
1176
+ name='allow_key_transform',
1177
+ full_name='protos.SeqAttGroupConfig.allow_key_transform',
1178
+ index=6,
1179
+ number=7,
1180
+ type=8,
1181
+ cpp_type=7,
1182
+ label=1,
1183
+ has_default_value=True,
1184
+ default_value=False,
1185
+ message_type=None,
1186
+ enum_type=None,
1187
+ containing_type=None,
1188
+ is_extension=False,
1189
+ extension_scope=None,
1190
+ options=None),
1191
+ _descriptor.FieldDescriptor(
1192
+ name='transform_dnn',
1193
+ full_name='protos.SeqAttGroupConfig.transform_dnn',
1194
+ index=7,
1195
+ number=8,
1196
+ type=8,
1197
+ cpp_type=7,
1198
+ label=1,
1199
+ has_default_value=True,
1200
+ default_value=False,
1201
+ message_type=None,
1202
+ enum_type=None,
1203
+ containing_type=None,
1204
+ is_extension=False,
1205
+ extension_scope=None,
1206
+ options=None),
1207
+ ],
1208
+ extensions=[],
1209
+ nested_types=[],
1210
+ enum_types=[],
1211
+ options=None,
1212
+ is_extendable=False,
1213
+ syntax='proto2',
1214
+ extension_ranges=[],
1215
+ oneofs=[],
1216
+ serialized_start=2069,
1217
+ serialized_end=2336,
1218
+ )
1219
+
1220
+ _SEQUENCECOMBINER.fields_by_name['attention'].message_type = _ATTENTIONCOMBINER
1221
+ _SEQUENCECOMBINER.fields_by_name[
1222
+ 'multi_head_attention'].message_type = _MULTIHEADATTENTIONCOMBINER
1223
+ _SEQUENCECOMBINER.fields_by_name[
1224
+ 'text_cnn'].message_type = easy__rec_dot_python_dot_protos_dot_layer__pb2._TEXTCNN
1225
+ _SEQUENCECOMBINER.oneofs_by_name['combiner'].fields.append(
1226
+ _SEQUENCECOMBINER.fields_by_name['attention'])
1227
+ _SEQUENCECOMBINER.fields_by_name[
1228
+ 'attention'].containing_oneof = _SEQUENCECOMBINER.oneofs_by_name['combiner']
1229
+ _SEQUENCECOMBINER.oneofs_by_name['combiner'].fields.append(
1230
+ _SEQUENCECOMBINER.fields_by_name['multi_head_attention'])
1231
+ _SEQUENCECOMBINER.fields_by_name[
1232
+ 'multi_head_attention'].containing_oneof = _SEQUENCECOMBINER.oneofs_by_name[
1233
+ 'combiner']
1234
+ _SEQUENCECOMBINER.oneofs_by_name['combiner'].fields.append(
1235
+ _SEQUENCECOMBINER.fields_by_name['text_cnn'])
1236
+ _SEQUENCECOMBINER.fields_by_name[
1237
+ 'text_cnn'].containing_oneof = _SEQUENCECOMBINER.oneofs_by_name['combiner']
1238
+ _FEATURECONFIG.fields_by_name[
1239
+ 'feature_type'].enum_type = _FEATURECONFIG_FEATURETYPE
1240
+ _FEATURECONFIG.fields_by_name[
1241
+ 'initializer'].message_type = easy__rec_dot_python_dot_protos_dot_hyperparams__pb2._INITIALIZER
1242
+ _FEATURECONFIG.fields_by_name[
1243
+ 'sequence_combiner'].message_type = _SEQUENCECOMBINER
1244
+ _FEATURECONFIG.fields_by_name[
1245
+ 'sub_feature_type'].enum_type = _FEATURECONFIG_FEATURETYPE
1246
+ _FEATURECONFIG.fields_by_name['ev_params'].message_type = _EVPARAMS
1247
+ _FEATURECONFIG_FEATURETYPE.containing_type = _FEATURECONFIG
1248
+ _FEATURECONFIG_FIELDTYPE.containing_type = _FEATURECONFIG
1249
+ _FEATURECONFIGV2.fields_by_name['features'].message_type = _FEATURECONFIG
1250
+ _FEATUREGROUPCONFIG.fields_by_name['wide_deep'].enum_type = _WIDEORDEEP
1251
+ _FEATUREGROUPCONFIG.fields_by_name[
1252
+ 'sequence_features'].message_type = _SEQATTGROUPCONFIG
1253
+ _SEQATTGROUPCONFIG.fields_by_name['seq_att_map'].message_type = _SEQATTMAP
1254
+ _SEQATTGROUPCONFIG.fields_by_name[
1255
+ 'seq_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
1256
+ DESCRIPTOR.message_types_by_name['AttentionCombiner'] = _ATTENTIONCOMBINER
1257
+ DESCRIPTOR.message_types_by_name[
1258
+ 'MultiHeadAttentionCombiner'] = _MULTIHEADATTENTIONCOMBINER
1259
+ DESCRIPTOR.message_types_by_name['SequenceCombiner'] = _SEQUENCECOMBINER
1260
+ DESCRIPTOR.message_types_by_name['EVParams'] = _EVPARAMS
1261
+ DESCRIPTOR.message_types_by_name['FeatureConfig'] = _FEATURECONFIG
1262
+ DESCRIPTOR.message_types_by_name['FeatureConfigV2'] = _FEATURECONFIGV2
1263
+ DESCRIPTOR.message_types_by_name['FeatureGroupConfig'] = _FEATUREGROUPCONFIG
1264
+ DESCRIPTOR.message_types_by_name['SeqAttMap'] = _SEQATTMAP
1265
+ DESCRIPTOR.message_types_by_name['SeqAttGroupConfig'] = _SEQATTGROUPCONFIG
1266
+ DESCRIPTOR.enum_types_by_name['WideOrDeep'] = _WIDEORDEEP
1267
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
1268
+
1269
+ AttentionCombiner = _reflection.GeneratedProtocolMessageType(
1270
+ 'AttentionCombiner',
1271
+ (_message.Message,),
1272
+ dict(
1273
+ DESCRIPTOR=_ATTENTIONCOMBINER,
1274
+ __module__='easy_rec.python.protos.feature_config_pb2'
1275
+ # @@protoc_insertion_point(class_scope:protos.AttentionCombiner)
1276
+ ))
1277
+ _sym_db.RegisterMessage(AttentionCombiner)
1278
+
1279
+ MultiHeadAttentionCombiner = _reflection.GeneratedProtocolMessageType(
1280
+ 'MultiHeadAttentionCombiner',
1281
+ (_message.Message,),
1282
+ dict(
1283
+ DESCRIPTOR=_MULTIHEADATTENTIONCOMBINER,
1284
+ __module__='easy_rec.python.protos.feature_config_pb2'
1285
+ # @@protoc_insertion_point(class_scope:protos.MultiHeadAttentionCombiner)
1286
+ ))
1287
+ _sym_db.RegisterMessage(MultiHeadAttentionCombiner)
1288
+
1289
+ SequenceCombiner = _reflection.GeneratedProtocolMessageType(
1290
+ 'SequenceCombiner',
1291
+ (_message.Message,),
1292
+ dict(
1293
+ DESCRIPTOR=_SEQUENCECOMBINER,
1294
+ __module__='easy_rec.python.protos.feature_config_pb2'
1295
+ # @@protoc_insertion_point(class_scope:protos.SequenceCombiner)
1296
+ ))
1297
+ _sym_db.RegisterMessage(SequenceCombiner)
1298
+
1299
+ EVParams = _reflection.GeneratedProtocolMessageType(
1300
+ 'EVParams',
1301
+ (_message.Message,),
1302
+ dict(
1303
+ DESCRIPTOR=_EVPARAMS,
1304
+ __module__='easy_rec.python.protos.feature_config_pb2'
1305
+ # @@protoc_insertion_point(class_scope:protos.EVParams)
1306
+ ))
1307
+ _sym_db.RegisterMessage(EVParams)
1308
+
1309
+ FeatureConfig = _reflection.GeneratedProtocolMessageType(
1310
+ 'FeatureConfig',
1311
+ (_message.Message,),
1312
+ dict(
1313
+ DESCRIPTOR=_FEATURECONFIG,
1314
+ __module__='easy_rec.python.protos.feature_config_pb2'
1315
+ # @@protoc_insertion_point(class_scope:protos.FeatureConfig)
1316
+ ))
1317
+ _sym_db.RegisterMessage(FeatureConfig)
1318
+
1319
+ FeatureConfigV2 = _reflection.GeneratedProtocolMessageType(
1320
+ 'FeatureConfigV2',
1321
+ (_message.Message,),
1322
+ dict(
1323
+ DESCRIPTOR=_FEATURECONFIGV2,
1324
+ __module__='easy_rec.python.protos.feature_config_pb2'
1325
+ # @@protoc_insertion_point(class_scope:protos.FeatureConfigV2)
1326
+ ))
1327
+ _sym_db.RegisterMessage(FeatureConfigV2)
1328
+
1329
+ FeatureGroupConfig = _reflection.GeneratedProtocolMessageType(
1330
+ 'FeatureGroupConfig',
1331
+ (_message.Message,),
1332
+ dict(
1333
+ DESCRIPTOR=_FEATUREGROUPCONFIG,
1334
+ __module__='easy_rec.python.protos.feature_config_pb2'
1335
+ # @@protoc_insertion_point(class_scope:protos.FeatureGroupConfig)
1336
+ ))
1337
+ _sym_db.RegisterMessage(FeatureGroupConfig)
1338
+
1339
+ SeqAttMap = _reflection.GeneratedProtocolMessageType(
1340
+ 'SeqAttMap',
1341
+ (_message.Message,),
1342
+ dict(
1343
+ DESCRIPTOR=_SEQATTMAP,
1344
+ __module__='easy_rec.python.protos.feature_config_pb2'
1345
+ # @@protoc_insertion_point(class_scope:protos.SeqAttMap)
1346
+ ))
1347
+ _sym_db.RegisterMessage(SeqAttMap)
1348
+
1349
+ SeqAttGroupConfig = _reflection.GeneratedProtocolMessageType(
1350
+ 'SeqAttGroupConfig',
1351
+ (_message.Message,),
1352
+ dict(
1353
+ DESCRIPTOR=_SEQATTGROUPCONFIG,
1354
+ __module__='easy_rec.python.protos.feature_config_pb2'
1355
+ # @@protoc_insertion_point(class_scope:protos.SeqAttGroupConfig)
1356
+ ))
1357
+ _sym_db.RegisterMessage(SeqAttGroupConfig)
1358
+
1359
+ # @@protoc_insertion_point(module_scope)