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,1713 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # source: easy_rec/python/protos/loss.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
+ _b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
14
+ # @@protoc_insertion_point(imports)
15
+
16
+ _sym_db = _symbol_database.Default()
17
+
18
+ DESCRIPTOR = _descriptor.FileDescriptor(
19
+ name='easy_rec/python/protos/loss.proto',
20
+ package='protos',
21
+ syntax='proto2',
22
+ serialized_pb=_b(
23
+ '\n!easy_rec/python/protos/loss.proto\x12\x06protos\"\x96\x06\n\x04Loss\x12#\n\tloss_type\x18\x01 \x02(\x0e\x32\x10.protos.LossType\x12\x11\n\x06weight\x18\x02 \x01(\x02:\x01\x31\x12\x11\n\tloss_name\x18\x03 \x01(\t\x12 \n\x11learn_loss_weight\x18\x04 \x01(\x08:\x05\x66\x61lse\x12\x35\n\x12\x66\x31_reweighted_loss\x18\x65 \x01(\x0b\x32\x17.protos.F1ReweighedLossH\x00\x12\x45\n\x0csoftmax_loss\x18\x66 \x01(\x0b\x32-.protos.SoftmaxCrossEntropyWithNegativeMiningH\x00\x12)\n\x0b\x63ircle_loss\x18g \x01(\x0b\x32\x12.protos.CircleLossH\x00\x12\x36\n\x0fmulti_simi_loss\x18h \x01(\x0b\x32\x1b.protos.MultiSimilarityLossH\x00\x12\x34\n\x11\x62inary_focal_loss\x18i \x01(\x0b\x32\x17.protos.BinaryFocalLossH\x00\x12-\n\rpairwise_loss\x18j \x01(\x0b\x32\x14.protos.PairwiseLossH\x00\x12\x38\n\x13pairwise_focal_loss\x18k \x01(\x0b\x32\x19.protos.PairwiseFocalLossH\x00\x12>\n\x16pairwise_logistic_loss\x18l \x01(\x0b\x32\x1c.protos.PairwiseLogisticLossH\x00\x12#\n\x08jrc_loss\x18m \x01(\x0b\x32\x0f.protos.JRCLossH\x00\x12\x38\n\x13pairwise_hinge_loss\x18n \x01(\x0b\x32\x19.protos.PairwiseHingeLossH\x00\x12\x36\n\x12listwise_rank_loss\x18o \x01(\x0b\x32\x18.protos.ListwiseRankLossH\x00\x12<\n\x15listwise_distill_loss\x18p \x01(\x0b\x32\x1b.protos.ListwiseDistillLossH\x00\x42\x0c\n\nloss_param\"\x94\x01\n%SoftmaxCrossEntropyWithNegativeMining\x12\x1c\n\x14num_negative_samples\x18\x01 \x02(\r\x12\x11\n\x06margin\x18\x02 \x02(\x02:\x01\x30\x12\x10\n\x05gamma\x18\x03 \x02(\x02:\x01\x31\x12(\n\x1d\x63oefficient_of_support_vector\x18\x04 \x02(\x02:\x01\x31\"5\n\nCircleLoss\x12\x14\n\x06margin\x18\x01 \x02(\x02:\x04\x30.25\x12\x11\n\x05gamma\x18\x02 \x02(\x02:\x02\x33\x32\"\\\n\x13MultiSimilarityLoss\x12\x10\n\x05\x61lpha\x18\x01 \x02(\x02:\x01\x32\x12\x10\n\x04\x62\x65ta\x18\x02 \x02(\x02:\x02\x35\x30\x12\x0f\n\x04lamb\x18\x03 \x02(\x02:\x01\x31\x12\x10\n\x03\x65ps\x18\x04 \x02(\x02:\x03\x30.1\"H\n\x0f\x46\x31ReweighedLoss\x12\x19\n\x0e\x66\x31_beta_square\x18\x01 \x02(\x02:\x01\x31\x12\x1a\n\x0flabel_smoothing\x18\x02 \x02(\x02:\x01\x30\"e\n\x0f\x42inaryFocalLoss\x12\x10\n\x05gamma\x18\x01 \x02(\x02:\x01\x32\x12\r\n\x05\x61lpha\x18\x02 \x01(\x02\x12\x15\n\nohem_ratio\x18\x03 \x01(\x02:\x01\x31\x12\x1a\n\x0flabel_smoothing\x18\x04 \x01(\x02:\x01\x30\"O\n\x0cPairwiseLoss\x12\x11\n\x06margin\x18\x01 \x02(\x02:\x01\x30\x12\x14\n\x0csession_name\x18\x02 \x01(\t\x12\x16\n\x0btemperature\x18\x03 \x01(\x02:\x01\x31\"\x92\x01\n\x11PairwiseFocalLoss\x12\x10\n\x05gamma\x18\x01 \x02(\x02:\x01\x32\x12\r\n\x05\x61lpha\x18\x02 \x01(\x02\x12\x17\n\x0chinge_margin\x18\x03 \x01(\x02:\x01\x31\x12\x14\n\x0csession_name\x18\x04 \x01(\t\x12\x15\n\nohem_ratio\x18\x05 \x01(\x02:\x01\x31\x12\x16\n\x0btemperature\x18\x06 \x01(\x02:\x01\x31\"\x92\x01\n\x14PairwiseLogisticLoss\x12\x16\n\x0btemperature\x18\x01 \x02(\x02:\x01\x31\x12\x14\n\x0csession_name\x18\x02 \x01(\t\x12\x14\n\x0chinge_margin\x18\x03 \x01(\x02\x12\x15\n\nohem_ratio\x18\x04 \x01(\x02:\x01\x31\x12\x1f\n\x10use_label_margin\x18\x05 \x01(\x08:\x05\x66\x61lse\"\xc7\x01\n\x11PairwiseHingeLoss\x12\x16\n\x0btemperature\x18\x01 \x02(\x02:\x01\x31\x12\x14\n\x0csession_name\x18\x02 \x01(\t\x12\x11\n\x06margin\x18\x03 \x01(\x02:\x01\x31\x12\x15\n\nohem_ratio\x18\x04 \x01(\x02:\x01\x31\x12\x1d\n\x0flabel_is_logits\x18\x05 \x01(\x08:\x04true\x12\x1e\n\x10use_label_margin\x18\x06 \x01(\x08:\x04true\x12\x1b\n\x0cuse_exponent\x18\x07 \x01(\x08:\x05\x66\x61lse\"w\n\x07JRCLoss\x12\x14\n\x0csession_name\x18\x01 \x02(\t\x12\x12\n\x05\x61lpha\x18\x02 \x01(\x02:\x03\x30.5\x12\x1d\n\x0fsame_label_loss\x18\x03 \x01(\x08:\x04true\x12#\n\x14loss_weight_strategy\x18\x04 \x02(\t:\x05\x66ixed\"\x93\x01\n\x10ListwiseRankLoss\x12\x16\n\x0btemperature\x18\x01 \x02(\x02:\x01\x31\x12\x14\n\x0csession_name\x18\x02 \x01(\t\x12\x14\n\x0ctransform_fn\x18\x03 \x01(\t\x12\x1e\n\x0flabel_is_logits\x18\x04 \x01(\x08:\x05\x66\x61lse\x12\x1b\n\x0cscale_logits\x18\x05 \x01(\x08:\x05\x66\x61lse\"\x99\x01\n\x13ListwiseDistillLoss\x12\x16\n\x0btemperature\x18\x01 \x02(\x02:\x01\x31\x12\x14\n\x0csession_name\x18\x02 \x01(\t\x12\x14\n\x0ctransform_fn\x18\x03 \x01(\t\x12!\n\x14label_clip_max_value\x18\x04 \x01(\x02:\x03\x35\x31\x32\x12\x1b\n\x0cscale_logits\x18\x05 \x01(\x08:\x05\x66\x61lse*\x81\x04\n\x08LossType\x12\x12\n\x0e\x43LASSIFICATION\x10\x00\x12\x0b\n\x07L2_LOSS\x10\x01\x12\x13\n\x0fSIGMOID_L2_LOSS\x10\x02\x12\x16\n\x12\x43ROSS_ENTROPY_LOSS\x10\x03\x12\x19\n\x15SOFTMAX_CROSS_ENTROPY\x10\x04\x12\x0f\n\x0b\x43IRCLE_LOSS\x10\x05\x12\x19\n\x15MULTI_SIMILARITY_LOSS\x10\x06\x12.\n*SOFTMAX_CROSS_ENTROPY_WITH_NEGATIVE_MINING\x10\x07\x12\x12\n\x0ePAIR_WISE_LOSS\x10\x08\x12\x16\n\x12\x46\x31_REWEIGHTED_LOSS\x10\t\x12\x15\n\x11\x42INARY_FOCAL_LOSS\x10\n\x12\x17\n\x13PAIRWISE_FOCAL_LOSS\x10\x0b\x12\x1a\n\x16PAIRWISE_LOGISTIC_LOSS\x10\x0c\x12\x17\n\x13PAIRWISE_HINGE_LOSS\x10\x11\x12\x0c\n\x08JRC_LOSS\x10\r\x12\x18\n\x14ORDER_CALIBRATE_LOSS\x10\x0e\x12\x1d\n\x19\x42INARY_CROSS_ENTROPY_LOSS\x10\x0f\x12\x16\n\x12KL_DIVERGENCE_LOSS\x10\x10\x12\x16\n\x12LISTWISE_RANK_LOSS\x10\x12\x12\x19\n\x15LISTWISE_DISTILL_LOSS\x10\x13\x12\r\n\tZILN_LOSS\x10\x14'
24
+ ))
25
+
26
+ _LOSSTYPE = _descriptor.EnumDescriptor(
27
+ name='LossType',
28
+ full_name='protos.LossType',
29
+ filename=None,
30
+ file=DESCRIPTOR,
31
+ values=[
32
+ _descriptor.EnumValueDescriptor(
33
+ name='CLASSIFICATION', index=0, number=0, options=None, type=None),
34
+ _descriptor.EnumValueDescriptor(
35
+ name='L2_LOSS', index=1, number=1, options=None, type=None),
36
+ _descriptor.EnumValueDescriptor(
37
+ name='SIGMOID_L2_LOSS', index=2, number=2, options=None, type=None),
38
+ _descriptor.EnumValueDescriptor(
39
+ name='CROSS_ENTROPY_LOSS',
40
+ index=3,
41
+ number=3,
42
+ options=None,
43
+ type=None),
44
+ _descriptor.EnumValueDescriptor(
45
+ name='SOFTMAX_CROSS_ENTROPY',
46
+ index=4,
47
+ number=4,
48
+ options=None,
49
+ type=None),
50
+ _descriptor.EnumValueDescriptor(
51
+ name='CIRCLE_LOSS', index=5, number=5, options=None, type=None),
52
+ _descriptor.EnumValueDescriptor(
53
+ name='MULTI_SIMILARITY_LOSS',
54
+ index=6,
55
+ number=6,
56
+ options=None,
57
+ type=None),
58
+ _descriptor.EnumValueDescriptor(
59
+ name='SOFTMAX_CROSS_ENTROPY_WITH_NEGATIVE_MINING',
60
+ index=7,
61
+ number=7,
62
+ options=None,
63
+ type=None),
64
+ _descriptor.EnumValueDescriptor(
65
+ name='PAIR_WISE_LOSS', index=8, number=8, options=None, type=None),
66
+ _descriptor.EnumValueDescriptor(
67
+ name='F1_REWEIGHTED_LOSS',
68
+ index=9,
69
+ number=9,
70
+ options=None,
71
+ type=None),
72
+ _descriptor.EnumValueDescriptor(
73
+ name='BINARY_FOCAL_LOSS',
74
+ index=10,
75
+ number=10,
76
+ options=None,
77
+ type=None),
78
+ _descriptor.EnumValueDescriptor(
79
+ name='PAIRWISE_FOCAL_LOSS',
80
+ index=11,
81
+ number=11,
82
+ options=None,
83
+ type=None),
84
+ _descriptor.EnumValueDescriptor(
85
+ name='PAIRWISE_LOGISTIC_LOSS',
86
+ index=12,
87
+ number=12,
88
+ options=None,
89
+ type=None),
90
+ _descriptor.EnumValueDescriptor(
91
+ name='PAIRWISE_HINGE_LOSS',
92
+ index=13,
93
+ number=17,
94
+ options=None,
95
+ type=None),
96
+ _descriptor.EnumValueDescriptor(
97
+ name='JRC_LOSS', index=14, number=13, options=None, type=None),
98
+ _descriptor.EnumValueDescriptor(
99
+ name='ORDER_CALIBRATE_LOSS',
100
+ index=15,
101
+ number=14,
102
+ options=None,
103
+ type=None),
104
+ _descriptor.EnumValueDescriptor(
105
+ name='BINARY_CROSS_ENTROPY_LOSS',
106
+ index=16,
107
+ number=15,
108
+ options=None,
109
+ type=None),
110
+ _descriptor.EnumValueDescriptor(
111
+ name='KL_DIVERGENCE_LOSS',
112
+ index=17,
113
+ number=16,
114
+ options=None,
115
+ type=None),
116
+ _descriptor.EnumValueDescriptor(
117
+ name='LISTWISE_RANK_LOSS',
118
+ index=18,
119
+ number=18,
120
+ options=None,
121
+ type=None),
122
+ _descriptor.EnumValueDescriptor(
123
+ name='LISTWISE_DISTILL_LOSS',
124
+ index=19,
125
+ number=19,
126
+ options=None,
127
+ type=None),
128
+ _descriptor.EnumValueDescriptor(
129
+ name='ZILN_LOSS', index=20, number=20, options=None, type=None),
130
+ ],
131
+ containing_type=None,
132
+ options=None,
133
+ serialized_start=2324,
134
+ serialized_end=2837,
135
+ )
136
+ _sym_db.RegisterEnumDescriptor(_LOSSTYPE)
137
+
138
+ LossType = enum_type_wrapper.EnumTypeWrapper(_LOSSTYPE)
139
+ CLASSIFICATION = 0
140
+ L2_LOSS = 1
141
+ SIGMOID_L2_LOSS = 2
142
+ CROSS_ENTROPY_LOSS = 3
143
+ SOFTMAX_CROSS_ENTROPY = 4
144
+ CIRCLE_LOSS = 5
145
+ MULTI_SIMILARITY_LOSS = 6
146
+ SOFTMAX_CROSS_ENTROPY_WITH_NEGATIVE_MINING = 7
147
+ PAIR_WISE_LOSS = 8
148
+ F1_REWEIGHTED_LOSS = 9
149
+ BINARY_FOCAL_LOSS = 10
150
+ PAIRWISE_FOCAL_LOSS = 11
151
+ PAIRWISE_LOGISTIC_LOSS = 12
152
+ PAIRWISE_HINGE_LOSS = 17
153
+ JRC_LOSS = 13
154
+ ORDER_CALIBRATE_LOSS = 14
155
+ BINARY_CROSS_ENTROPY_LOSS = 15
156
+ KL_DIVERGENCE_LOSS = 16
157
+ LISTWISE_RANK_LOSS = 18
158
+ LISTWISE_DISTILL_LOSS = 19
159
+ ZILN_LOSS = 20
160
+
161
+ _LOSS = _descriptor.Descriptor(
162
+ name='Loss',
163
+ full_name='protos.Loss',
164
+ filename=None,
165
+ file=DESCRIPTOR,
166
+ containing_type=None,
167
+ fields=[
168
+ _descriptor.FieldDescriptor(
169
+ name='loss_type',
170
+ full_name='protos.Loss.loss_type',
171
+ index=0,
172
+ number=1,
173
+ type=14,
174
+ cpp_type=8,
175
+ label=2,
176
+ has_default_value=False,
177
+ default_value=0,
178
+ message_type=None,
179
+ enum_type=None,
180
+ containing_type=None,
181
+ is_extension=False,
182
+ extension_scope=None,
183
+ options=None),
184
+ _descriptor.FieldDescriptor(
185
+ name='weight',
186
+ full_name='protos.Loss.weight',
187
+ index=1,
188
+ number=2,
189
+ type=2,
190
+ cpp_type=6,
191
+ label=1,
192
+ has_default_value=True,
193
+ default_value=float(1),
194
+ message_type=None,
195
+ enum_type=None,
196
+ containing_type=None,
197
+ is_extension=False,
198
+ extension_scope=None,
199
+ options=None),
200
+ _descriptor.FieldDescriptor(
201
+ name='loss_name',
202
+ full_name='protos.Loss.loss_name',
203
+ index=2,
204
+ number=3,
205
+ type=9,
206
+ cpp_type=9,
207
+ label=1,
208
+ has_default_value=False,
209
+ default_value=_b('').decode('utf-8'),
210
+ message_type=None,
211
+ enum_type=None,
212
+ containing_type=None,
213
+ is_extension=False,
214
+ extension_scope=None,
215
+ options=None),
216
+ _descriptor.FieldDescriptor(
217
+ name='learn_loss_weight',
218
+ full_name='protos.Loss.learn_loss_weight',
219
+ index=3,
220
+ number=4,
221
+ type=8,
222
+ cpp_type=7,
223
+ label=1,
224
+ has_default_value=True,
225
+ default_value=False,
226
+ message_type=None,
227
+ enum_type=None,
228
+ containing_type=None,
229
+ is_extension=False,
230
+ extension_scope=None,
231
+ options=None),
232
+ _descriptor.FieldDescriptor(
233
+ name='f1_reweighted_loss',
234
+ full_name='protos.Loss.f1_reweighted_loss',
235
+ index=4,
236
+ number=101,
237
+ type=11,
238
+ cpp_type=10,
239
+ label=1,
240
+ has_default_value=False,
241
+ default_value=None,
242
+ message_type=None,
243
+ enum_type=None,
244
+ containing_type=None,
245
+ is_extension=False,
246
+ extension_scope=None,
247
+ options=None),
248
+ _descriptor.FieldDescriptor(
249
+ name='softmax_loss',
250
+ full_name='protos.Loss.softmax_loss',
251
+ index=5,
252
+ number=102,
253
+ type=11,
254
+ cpp_type=10,
255
+ label=1,
256
+ has_default_value=False,
257
+ default_value=None,
258
+ message_type=None,
259
+ enum_type=None,
260
+ containing_type=None,
261
+ is_extension=False,
262
+ extension_scope=None,
263
+ options=None),
264
+ _descriptor.FieldDescriptor(
265
+ name='circle_loss',
266
+ full_name='protos.Loss.circle_loss',
267
+ index=6,
268
+ number=103,
269
+ type=11,
270
+ cpp_type=10,
271
+ label=1,
272
+ has_default_value=False,
273
+ default_value=None,
274
+ message_type=None,
275
+ enum_type=None,
276
+ containing_type=None,
277
+ is_extension=False,
278
+ extension_scope=None,
279
+ options=None),
280
+ _descriptor.FieldDescriptor(
281
+ name='multi_simi_loss',
282
+ full_name='protos.Loss.multi_simi_loss',
283
+ index=7,
284
+ number=104,
285
+ type=11,
286
+ cpp_type=10,
287
+ label=1,
288
+ has_default_value=False,
289
+ default_value=None,
290
+ message_type=None,
291
+ enum_type=None,
292
+ containing_type=None,
293
+ is_extension=False,
294
+ extension_scope=None,
295
+ options=None),
296
+ _descriptor.FieldDescriptor(
297
+ name='binary_focal_loss',
298
+ full_name='protos.Loss.binary_focal_loss',
299
+ index=8,
300
+ number=105,
301
+ type=11,
302
+ cpp_type=10,
303
+ label=1,
304
+ has_default_value=False,
305
+ default_value=None,
306
+ message_type=None,
307
+ enum_type=None,
308
+ containing_type=None,
309
+ is_extension=False,
310
+ extension_scope=None,
311
+ options=None),
312
+ _descriptor.FieldDescriptor(
313
+ name='pairwise_loss',
314
+ full_name='protos.Loss.pairwise_loss',
315
+ index=9,
316
+ number=106,
317
+ type=11,
318
+ cpp_type=10,
319
+ label=1,
320
+ has_default_value=False,
321
+ default_value=None,
322
+ message_type=None,
323
+ enum_type=None,
324
+ containing_type=None,
325
+ is_extension=False,
326
+ extension_scope=None,
327
+ options=None),
328
+ _descriptor.FieldDescriptor(
329
+ name='pairwise_focal_loss',
330
+ full_name='protos.Loss.pairwise_focal_loss',
331
+ index=10,
332
+ number=107,
333
+ type=11,
334
+ cpp_type=10,
335
+ label=1,
336
+ has_default_value=False,
337
+ default_value=None,
338
+ message_type=None,
339
+ enum_type=None,
340
+ containing_type=None,
341
+ is_extension=False,
342
+ extension_scope=None,
343
+ options=None),
344
+ _descriptor.FieldDescriptor(
345
+ name='pairwise_logistic_loss',
346
+ full_name='protos.Loss.pairwise_logistic_loss',
347
+ index=11,
348
+ number=108,
349
+ type=11,
350
+ cpp_type=10,
351
+ label=1,
352
+ has_default_value=False,
353
+ default_value=None,
354
+ message_type=None,
355
+ enum_type=None,
356
+ containing_type=None,
357
+ is_extension=False,
358
+ extension_scope=None,
359
+ options=None),
360
+ _descriptor.FieldDescriptor(
361
+ name='jrc_loss',
362
+ full_name='protos.Loss.jrc_loss',
363
+ index=12,
364
+ number=109,
365
+ type=11,
366
+ cpp_type=10,
367
+ label=1,
368
+ has_default_value=False,
369
+ default_value=None,
370
+ message_type=None,
371
+ enum_type=None,
372
+ containing_type=None,
373
+ is_extension=False,
374
+ extension_scope=None,
375
+ options=None),
376
+ _descriptor.FieldDescriptor(
377
+ name='pairwise_hinge_loss',
378
+ full_name='protos.Loss.pairwise_hinge_loss',
379
+ index=13,
380
+ number=110,
381
+ type=11,
382
+ cpp_type=10,
383
+ label=1,
384
+ has_default_value=False,
385
+ default_value=None,
386
+ message_type=None,
387
+ enum_type=None,
388
+ containing_type=None,
389
+ is_extension=False,
390
+ extension_scope=None,
391
+ options=None),
392
+ _descriptor.FieldDescriptor(
393
+ name='listwise_rank_loss',
394
+ full_name='protos.Loss.listwise_rank_loss',
395
+ index=14,
396
+ number=111,
397
+ type=11,
398
+ cpp_type=10,
399
+ label=1,
400
+ has_default_value=False,
401
+ default_value=None,
402
+ message_type=None,
403
+ enum_type=None,
404
+ containing_type=None,
405
+ is_extension=False,
406
+ extension_scope=None,
407
+ options=None),
408
+ _descriptor.FieldDescriptor(
409
+ name='listwise_distill_loss',
410
+ full_name='protos.Loss.listwise_distill_loss',
411
+ index=15,
412
+ number=112,
413
+ type=11,
414
+ cpp_type=10,
415
+ label=1,
416
+ has_default_value=False,
417
+ default_value=None,
418
+ message_type=None,
419
+ enum_type=None,
420
+ containing_type=None,
421
+ is_extension=False,
422
+ extension_scope=None,
423
+ options=None),
424
+ ],
425
+ extensions=[],
426
+ nested_types=[],
427
+ enum_types=[],
428
+ options=None,
429
+ is_extendable=False,
430
+ syntax='proto2',
431
+ extension_ranges=[],
432
+ oneofs=[
433
+ _descriptor.OneofDescriptor(
434
+ name='loss_param',
435
+ full_name='protos.Loss.loss_param',
436
+ index=0,
437
+ containing_type=None,
438
+ fields=[]),
439
+ ],
440
+ serialized_start=46,
441
+ serialized_end=836,
442
+ )
443
+
444
+ _SOFTMAXCROSSENTROPYWITHNEGATIVEMINING = _descriptor.Descriptor(
445
+ name='SoftmaxCrossEntropyWithNegativeMining',
446
+ full_name='protos.SoftmaxCrossEntropyWithNegativeMining',
447
+ filename=None,
448
+ file=DESCRIPTOR,
449
+ containing_type=None,
450
+ fields=[
451
+ _descriptor.FieldDescriptor(
452
+ name='num_negative_samples',
453
+ full_name='protos.SoftmaxCrossEntropyWithNegativeMining.num_negative_samples',
454
+ index=0,
455
+ number=1,
456
+ type=13,
457
+ cpp_type=3,
458
+ label=2,
459
+ has_default_value=False,
460
+ default_value=0,
461
+ message_type=None,
462
+ enum_type=None,
463
+ containing_type=None,
464
+ is_extension=False,
465
+ extension_scope=None,
466
+ options=None),
467
+ _descriptor.FieldDescriptor(
468
+ name='margin',
469
+ full_name='protos.SoftmaxCrossEntropyWithNegativeMining.margin',
470
+ index=1,
471
+ number=2,
472
+ type=2,
473
+ cpp_type=6,
474
+ label=2,
475
+ has_default_value=True,
476
+ default_value=float(0),
477
+ message_type=None,
478
+ enum_type=None,
479
+ containing_type=None,
480
+ is_extension=False,
481
+ extension_scope=None,
482
+ options=None),
483
+ _descriptor.FieldDescriptor(
484
+ name='gamma',
485
+ full_name='protos.SoftmaxCrossEntropyWithNegativeMining.gamma',
486
+ index=2,
487
+ number=3,
488
+ type=2,
489
+ cpp_type=6,
490
+ label=2,
491
+ has_default_value=True,
492
+ default_value=float(1),
493
+ message_type=None,
494
+ enum_type=None,
495
+ containing_type=None,
496
+ is_extension=False,
497
+ extension_scope=None,
498
+ options=None),
499
+ _descriptor.FieldDescriptor(
500
+ name='coefficient_of_support_vector',
501
+ full_name='protos.SoftmaxCrossEntropyWithNegativeMining.coefficient_of_support_vector',
502
+ index=3,
503
+ number=4,
504
+ type=2,
505
+ cpp_type=6,
506
+ label=2,
507
+ has_default_value=True,
508
+ default_value=float(1),
509
+ message_type=None,
510
+ enum_type=None,
511
+ containing_type=None,
512
+ is_extension=False,
513
+ extension_scope=None,
514
+ options=None),
515
+ ],
516
+ extensions=[],
517
+ nested_types=[],
518
+ enum_types=[],
519
+ options=None,
520
+ is_extendable=False,
521
+ syntax='proto2',
522
+ extension_ranges=[],
523
+ oneofs=[],
524
+ serialized_start=839,
525
+ serialized_end=987,
526
+ )
527
+
528
+ _CIRCLELOSS = _descriptor.Descriptor(
529
+ name='CircleLoss',
530
+ full_name='protos.CircleLoss',
531
+ filename=None,
532
+ file=DESCRIPTOR,
533
+ containing_type=None,
534
+ fields=[
535
+ _descriptor.FieldDescriptor(
536
+ name='margin',
537
+ full_name='protos.CircleLoss.margin',
538
+ index=0,
539
+ number=1,
540
+ type=2,
541
+ cpp_type=6,
542
+ label=2,
543
+ has_default_value=True,
544
+ default_value=float(0.25),
545
+ message_type=None,
546
+ enum_type=None,
547
+ containing_type=None,
548
+ is_extension=False,
549
+ extension_scope=None,
550
+ options=None),
551
+ _descriptor.FieldDescriptor(
552
+ name='gamma',
553
+ full_name='protos.CircleLoss.gamma',
554
+ index=1,
555
+ number=2,
556
+ type=2,
557
+ cpp_type=6,
558
+ label=2,
559
+ has_default_value=True,
560
+ default_value=float(32),
561
+ message_type=None,
562
+ enum_type=None,
563
+ containing_type=None,
564
+ is_extension=False,
565
+ extension_scope=None,
566
+ options=None),
567
+ ],
568
+ extensions=[],
569
+ nested_types=[],
570
+ enum_types=[],
571
+ options=None,
572
+ is_extendable=False,
573
+ syntax='proto2',
574
+ extension_ranges=[],
575
+ oneofs=[],
576
+ serialized_start=989,
577
+ serialized_end=1042,
578
+ )
579
+
580
+ _MULTISIMILARITYLOSS = _descriptor.Descriptor(
581
+ name='MultiSimilarityLoss',
582
+ full_name='protos.MultiSimilarityLoss',
583
+ filename=None,
584
+ file=DESCRIPTOR,
585
+ containing_type=None,
586
+ fields=[
587
+ _descriptor.FieldDescriptor(
588
+ name='alpha',
589
+ full_name='protos.MultiSimilarityLoss.alpha',
590
+ index=0,
591
+ number=1,
592
+ type=2,
593
+ cpp_type=6,
594
+ label=2,
595
+ has_default_value=True,
596
+ default_value=float(2),
597
+ message_type=None,
598
+ enum_type=None,
599
+ containing_type=None,
600
+ is_extension=False,
601
+ extension_scope=None,
602
+ options=None),
603
+ _descriptor.FieldDescriptor(
604
+ name='beta',
605
+ full_name='protos.MultiSimilarityLoss.beta',
606
+ index=1,
607
+ number=2,
608
+ type=2,
609
+ cpp_type=6,
610
+ label=2,
611
+ has_default_value=True,
612
+ default_value=float(50),
613
+ message_type=None,
614
+ enum_type=None,
615
+ containing_type=None,
616
+ is_extension=False,
617
+ extension_scope=None,
618
+ options=None),
619
+ _descriptor.FieldDescriptor(
620
+ name='lamb',
621
+ full_name='protos.MultiSimilarityLoss.lamb',
622
+ index=2,
623
+ number=3,
624
+ type=2,
625
+ cpp_type=6,
626
+ label=2,
627
+ has_default_value=True,
628
+ default_value=float(1),
629
+ message_type=None,
630
+ enum_type=None,
631
+ containing_type=None,
632
+ is_extension=False,
633
+ extension_scope=None,
634
+ options=None),
635
+ _descriptor.FieldDescriptor(
636
+ name='eps',
637
+ full_name='protos.MultiSimilarityLoss.eps',
638
+ index=3,
639
+ number=4,
640
+ type=2,
641
+ cpp_type=6,
642
+ label=2,
643
+ has_default_value=True,
644
+ default_value=float(0.1),
645
+ message_type=None,
646
+ enum_type=None,
647
+ containing_type=None,
648
+ is_extension=False,
649
+ extension_scope=None,
650
+ options=None),
651
+ ],
652
+ extensions=[],
653
+ nested_types=[],
654
+ enum_types=[],
655
+ options=None,
656
+ is_extendable=False,
657
+ syntax='proto2',
658
+ extension_ranges=[],
659
+ oneofs=[],
660
+ serialized_start=1044,
661
+ serialized_end=1136,
662
+ )
663
+
664
+ _F1REWEIGHEDLOSS = _descriptor.Descriptor(
665
+ name='F1ReweighedLoss',
666
+ full_name='protos.F1ReweighedLoss',
667
+ filename=None,
668
+ file=DESCRIPTOR,
669
+ containing_type=None,
670
+ fields=[
671
+ _descriptor.FieldDescriptor(
672
+ name='f1_beta_square',
673
+ full_name='protos.F1ReweighedLoss.f1_beta_square',
674
+ index=0,
675
+ number=1,
676
+ type=2,
677
+ cpp_type=6,
678
+ label=2,
679
+ has_default_value=True,
680
+ default_value=float(1),
681
+ message_type=None,
682
+ enum_type=None,
683
+ containing_type=None,
684
+ is_extension=False,
685
+ extension_scope=None,
686
+ options=None),
687
+ _descriptor.FieldDescriptor(
688
+ name='label_smoothing',
689
+ full_name='protos.F1ReweighedLoss.label_smoothing',
690
+ index=1,
691
+ number=2,
692
+ type=2,
693
+ cpp_type=6,
694
+ label=2,
695
+ has_default_value=True,
696
+ default_value=float(0),
697
+ message_type=None,
698
+ enum_type=None,
699
+ containing_type=None,
700
+ is_extension=False,
701
+ extension_scope=None,
702
+ options=None),
703
+ ],
704
+ extensions=[],
705
+ nested_types=[],
706
+ enum_types=[],
707
+ options=None,
708
+ is_extendable=False,
709
+ syntax='proto2',
710
+ extension_ranges=[],
711
+ oneofs=[],
712
+ serialized_start=1138,
713
+ serialized_end=1210,
714
+ )
715
+
716
+ _BINARYFOCALLOSS = _descriptor.Descriptor(
717
+ name='BinaryFocalLoss',
718
+ full_name='protos.BinaryFocalLoss',
719
+ filename=None,
720
+ file=DESCRIPTOR,
721
+ containing_type=None,
722
+ fields=[
723
+ _descriptor.FieldDescriptor(
724
+ name='gamma',
725
+ full_name='protos.BinaryFocalLoss.gamma',
726
+ index=0,
727
+ number=1,
728
+ type=2,
729
+ cpp_type=6,
730
+ label=2,
731
+ has_default_value=True,
732
+ default_value=float(2),
733
+ message_type=None,
734
+ enum_type=None,
735
+ containing_type=None,
736
+ is_extension=False,
737
+ extension_scope=None,
738
+ options=None),
739
+ _descriptor.FieldDescriptor(
740
+ name='alpha',
741
+ full_name='protos.BinaryFocalLoss.alpha',
742
+ index=1,
743
+ number=2,
744
+ type=2,
745
+ cpp_type=6,
746
+ label=1,
747
+ has_default_value=False,
748
+ default_value=float(0),
749
+ message_type=None,
750
+ enum_type=None,
751
+ containing_type=None,
752
+ is_extension=False,
753
+ extension_scope=None,
754
+ options=None),
755
+ _descriptor.FieldDescriptor(
756
+ name='ohem_ratio',
757
+ full_name='protos.BinaryFocalLoss.ohem_ratio',
758
+ index=2,
759
+ number=3,
760
+ type=2,
761
+ cpp_type=6,
762
+ label=1,
763
+ has_default_value=True,
764
+ default_value=float(1),
765
+ message_type=None,
766
+ enum_type=None,
767
+ containing_type=None,
768
+ is_extension=False,
769
+ extension_scope=None,
770
+ options=None),
771
+ _descriptor.FieldDescriptor(
772
+ name='label_smoothing',
773
+ full_name='protos.BinaryFocalLoss.label_smoothing',
774
+ index=3,
775
+ number=4,
776
+ type=2,
777
+ cpp_type=6,
778
+ label=1,
779
+ has_default_value=True,
780
+ default_value=float(0),
781
+ message_type=None,
782
+ enum_type=None,
783
+ containing_type=None,
784
+ is_extension=False,
785
+ extension_scope=None,
786
+ options=None),
787
+ ],
788
+ extensions=[],
789
+ nested_types=[],
790
+ enum_types=[],
791
+ options=None,
792
+ is_extendable=False,
793
+ syntax='proto2',
794
+ extension_ranges=[],
795
+ oneofs=[],
796
+ serialized_start=1212,
797
+ serialized_end=1313,
798
+ )
799
+
800
+ _PAIRWISELOSS = _descriptor.Descriptor(
801
+ name='PairwiseLoss',
802
+ full_name='protos.PairwiseLoss',
803
+ filename=None,
804
+ file=DESCRIPTOR,
805
+ containing_type=None,
806
+ fields=[
807
+ _descriptor.FieldDescriptor(
808
+ name='margin',
809
+ full_name='protos.PairwiseLoss.margin',
810
+ index=0,
811
+ number=1,
812
+ type=2,
813
+ cpp_type=6,
814
+ label=2,
815
+ has_default_value=True,
816
+ default_value=float(0),
817
+ message_type=None,
818
+ enum_type=None,
819
+ containing_type=None,
820
+ is_extension=False,
821
+ extension_scope=None,
822
+ options=None),
823
+ _descriptor.FieldDescriptor(
824
+ name='session_name',
825
+ full_name='protos.PairwiseLoss.session_name',
826
+ index=1,
827
+ number=2,
828
+ type=9,
829
+ cpp_type=9,
830
+ label=1,
831
+ has_default_value=False,
832
+ default_value=_b('').decode('utf-8'),
833
+ message_type=None,
834
+ enum_type=None,
835
+ containing_type=None,
836
+ is_extension=False,
837
+ extension_scope=None,
838
+ options=None),
839
+ _descriptor.FieldDescriptor(
840
+ name='temperature',
841
+ full_name='protos.PairwiseLoss.temperature',
842
+ index=2,
843
+ number=3,
844
+ type=2,
845
+ cpp_type=6,
846
+ label=1,
847
+ has_default_value=True,
848
+ default_value=float(1),
849
+ message_type=None,
850
+ enum_type=None,
851
+ containing_type=None,
852
+ is_extension=False,
853
+ extension_scope=None,
854
+ options=None),
855
+ ],
856
+ extensions=[],
857
+ nested_types=[],
858
+ enum_types=[],
859
+ options=None,
860
+ is_extendable=False,
861
+ syntax='proto2',
862
+ extension_ranges=[],
863
+ oneofs=[],
864
+ serialized_start=1315,
865
+ serialized_end=1394,
866
+ )
867
+
868
+ _PAIRWISEFOCALLOSS = _descriptor.Descriptor(
869
+ name='PairwiseFocalLoss',
870
+ full_name='protos.PairwiseFocalLoss',
871
+ filename=None,
872
+ file=DESCRIPTOR,
873
+ containing_type=None,
874
+ fields=[
875
+ _descriptor.FieldDescriptor(
876
+ name='gamma',
877
+ full_name='protos.PairwiseFocalLoss.gamma',
878
+ index=0,
879
+ number=1,
880
+ type=2,
881
+ cpp_type=6,
882
+ label=2,
883
+ has_default_value=True,
884
+ default_value=float(2),
885
+ message_type=None,
886
+ enum_type=None,
887
+ containing_type=None,
888
+ is_extension=False,
889
+ extension_scope=None,
890
+ options=None),
891
+ _descriptor.FieldDescriptor(
892
+ name='alpha',
893
+ full_name='protos.PairwiseFocalLoss.alpha',
894
+ index=1,
895
+ number=2,
896
+ type=2,
897
+ cpp_type=6,
898
+ label=1,
899
+ has_default_value=False,
900
+ default_value=float(0),
901
+ message_type=None,
902
+ enum_type=None,
903
+ containing_type=None,
904
+ is_extension=False,
905
+ extension_scope=None,
906
+ options=None),
907
+ _descriptor.FieldDescriptor(
908
+ name='hinge_margin',
909
+ full_name='protos.PairwiseFocalLoss.hinge_margin',
910
+ index=2,
911
+ number=3,
912
+ type=2,
913
+ cpp_type=6,
914
+ label=1,
915
+ has_default_value=True,
916
+ default_value=float(1),
917
+ message_type=None,
918
+ enum_type=None,
919
+ containing_type=None,
920
+ is_extension=False,
921
+ extension_scope=None,
922
+ options=None),
923
+ _descriptor.FieldDescriptor(
924
+ name='session_name',
925
+ full_name='protos.PairwiseFocalLoss.session_name',
926
+ index=3,
927
+ number=4,
928
+ type=9,
929
+ cpp_type=9,
930
+ label=1,
931
+ has_default_value=False,
932
+ default_value=_b('').decode('utf-8'),
933
+ message_type=None,
934
+ enum_type=None,
935
+ containing_type=None,
936
+ is_extension=False,
937
+ extension_scope=None,
938
+ options=None),
939
+ _descriptor.FieldDescriptor(
940
+ name='ohem_ratio',
941
+ full_name='protos.PairwiseFocalLoss.ohem_ratio',
942
+ index=4,
943
+ number=5,
944
+ type=2,
945
+ cpp_type=6,
946
+ label=1,
947
+ has_default_value=True,
948
+ default_value=float(1),
949
+ message_type=None,
950
+ enum_type=None,
951
+ containing_type=None,
952
+ is_extension=False,
953
+ extension_scope=None,
954
+ options=None),
955
+ _descriptor.FieldDescriptor(
956
+ name='temperature',
957
+ full_name='protos.PairwiseFocalLoss.temperature',
958
+ index=5,
959
+ number=6,
960
+ type=2,
961
+ cpp_type=6,
962
+ label=1,
963
+ has_default_value=True,
964
+ default_value=float(1),
965
+ message_type=None,
966
+ enum_type=None,
967
+ containing_type=None,
968
+ is_extension=False,
969
+ extension_scope=None,
970
+ options=None),
971
+ ],
972
+ extensions=[],
973
+ nested_types=[],
974
+ enum_types=[],
975
+ options=None,
976
+ is_extendable=False,
977
+ syntax='proto2',
978
+ extension_ranges=[],
979
+ oneofs=[],
980
+ serialized_start=1397,
981
+ serialized_end=1543,
982
+ )
983
+
984
+ _PAIRWISELOGISTICLOSS = _descriptor.Descriptor(
985
+ name='PairwiseLogisticLoss',
986
+ full_name='protos.PairwiseLogisticLoss',
987
+ filename=None,
988
+ file=DESCRIPTOR,
989
+ containing_type=None,
990
+ fields=[
991
+ _descriptor.FieldDescriptor(
992
+ name='temperature',
993
+ full_name='protos.PairwiseLogisticLoss.temperature',
994
+ index=0,
995
+ number=1,
996
+ type=2,
997
+ cpp_type=6,
998
+ label=2,
999
+ has_default_value=True,
1000
+ default_value=float(1),
1001
+ message_type=None,
1002
+ enum_type=None,
1003
+ containing_type=None,
1004
+ is_extension=False,
1005
+ extension_scope=None,
1006
+ options=None),
1007
+ _descriptor.FieldDescriptor(
1008
+ name='session_name',
1009
+ full_name='protos.PairwiseLogisticLoss.session_name',
1010
+ index=1,
1011
+ number=2,
1012
+ type=9,
1013
+ cpp_type=9,
1014
+ label=1,
1015
+ has_default_value=False,
1016
+ default_value=_b('').decode('utf-8'),
1017
+ message_type=None,
1018
+ enum_type=None,
1019
+ containing_type=None,
1020
+ is_extension=False,
1021
+ extension_scope=None,
1022
+ options=None),
1023
+ _descriptor.FieldDescriptor(
1024
+ name='hinge_margin',
1025
+ full_name='protos.PairwiseLogisticLoss.hinge_margin',
1026
+ index=2,
1027
+ number=3,
1028
+ type=2,
1029
+ cpp_type=6,
1030
+ label=1,
1031
+ has_default_value=False,
1032
+ default_value=float(0),
1033
+ message_type=None,
1034
+ enum_type=None,
1035
+ containing_type=None,
1036
+ is_extension=False,
1037
+ extension_scope=None,
1038
+ options=None),
1039
+ _descriptor.FieldDescriptor(
1040
+ name='ohem_ratio',
1041
+ full_name='protos.PairwiseLogisticLoss.ohem_ratio',
1042
+ index=3,
1043
+ number=4,
1044
+ type=2,
1045
+ cpp_type=6,
1046
+ label=1,
1047
+ has_default_value=True,
1048
+ default_value=float(1),
1049
+ message_type=None,
1050
+ enum_type=None,
1051
+ containing_type=None,
1052
+ is_extension=False,
1053
+ extension_scope=None,
1054
+ options=None),
1055
+ _descriptor.FieldDescriptor(
1056
+ name='use_label_margin',
1057
+ full_name='protos.PairwiseLogisticLoss.use_label_margin',
1058
+ index=4,
1059
+ number=5,
1060
+ type=8,
1061
+ cpp_type=7,
1062
+ label=1,
1063
+ has_default_value=True,
1064
+ default_value=False,
1065
+ message_type=None,
1066
+ enum_type=None,
1067
+ containing_type=None,
1068
+ is_extension=False,
1069
+ extension_scope=None,
1070
+ options=None),
1071
+ ],
1072
+ extensions=[],
1073
+ nested_types=[],
1074
+ enum_types=[],
1075
+ options=None,
1076
+ is_extendable=False,
1077
+ syntax='proto2',
1078
+ extension_ranges=[],
1079
+ oneofs=[],
1080
+ serialized_start=1546,
1081
+ serialized_end=1692,
1082
+ )
1083
+
1084
+ _PAIRWISEHINGELOSS = _descriptor.Descriptor(
1085
+ name='PairwiseHingeLoss',
1086
+ full_name='protos.PairwiseHingeLoss',
1087
+ filename=None,
1088
+ file=DESCRIPTOR,
1089
+ containing_type=None,
1090
+ fields=[
1091
+ _descriptor.FieldDescriptor(
1092
+ name='temperature',
1093
+ full_name='protos.PairwiseHingeLoss.temperature',
1094
+ index=0,
1095
+ number=1,
1096
+ type=2,
1097
+ cpp_type=6,
1098
+ label=2,
1099
+ has_default_value=True,
1100
+ default_value=float(1),
1101
+ message_type=None,
1102
+ enum_type=None,
1103
+ containing_type=None,
1104
+ is_extension=False,
1105
+ extension_scope=None,
1106
+ options=None),
1107
+ _descriptor.FieldDescriptor(
1108
+ name='session_name',
1109
+ full_name='protos.PairwiseHingeLoss.session_name',
1110
+ index=1,
1111
+ number=2,
1112
+ type=9,
1113
+ cpp_type=9,
1114
+ label=1,
1115
+ has_default_value=False,
1116
+ default_value=_b('').decode('utf-8'),
1117
+ message_type=None,
1118
+ enum_type=None,
1119
+ containing_type=None,
1120
+ is_extension=False,
1121
+ extension_scope=None,
1122
+ options=None),
1123
+ _descriptor.FieldDescriptor(
1124
+ name='margin',
1125
+ full_name='protos.PairwiseHingeLoss.margin',
1126
+ index=2,
1127
+ number=3,
1128
+ type=2,
1129
+ cpp_type=6,
1130
+ label=1,
1131
+ has_default_value=True,
1132
+ default_value=float(1),
1133
+ message_type=None,
1134
+ enum_type=None,
1135
+ containing_type=None,
1136
+ is_extension=False,
1137
+ extension_scope=None,
1138
+ options=None),
1139
+ _descriptor.FieldDescriptor(
1140
+ name='ohem_ratio',
1141
+ full_name='protos.PairwiseHingeLoss.ohem_ratio',
1142
+ index=3,
1143
+ number=4,
1144
+ type=2,
1145
+ cpp_type=6,
1146
+ label=1,
1147
+ has_default_value=True,
1148
+ default_value=float(1),
1149
+ message_type=None,
1150
+ enum_type=None,
1151
+ containing_type=None,
1152
+ is_extension=False,
1153
+ extension_scope=None,
1154
+ options=None),
1155
+ _descriptor.FieldDescriptor(
1156
+ name='label_is_logits',
1157
+ full_name='protos.PairwiseHingeLoss.label_is_logits',
1158
+ index=4,
1159
+ number=5,
1160
+ type=8,
1161
+ cpp_type=7,
1162
+ label=1,
1163
+ has_default_value=True,
1164
+ default_value=True,
1165
+ message_type=None,
1166
+ enum_type=None,
1167
+ containing_type=None,
1168
+ is_extension=False,
1169
+ extension_scope=None,
1170
+ options=None),
1171
+ _descriptor.FieldDescriptor(
1172
+ name='use_label_margin',
1173
+ full_name='protos.PairwiseHingeLoss.use_label_margin',
1174
+ index=5,
1175
+ number=6,
1176
+ type=8,
1177
+ cpp_type=7,
1178
+ label=1,
1179
+ has_default_value=True,
1180
+ default_value=True,
1181
+ message_type=None,
1182
+ enum_type=None,
1183
+ containing_type=None,
1184
+ is_extension=False,
1185
+ extension_scope=None,
1186
+ options=None),
1187
+ _descriptor.FieldDescriptor(
1188
+ name='use_exponent',
1189
+ full_name='protos.PairwiseHingeLoss.use_exponent',
1190
+ index=6,
1191
+ number=7,
1192
+ type=8,
1193
+ cpp_type=7,
1194
+ label=1,
1195
+ has_default_value=True,
1196
+ default_value=False,
1197
+ message_type=None,
1198
+ enum_type=None,
1199
+ containing_type=None,
1200
+ is_extension=False,
1201
+ extension_scope=None,
1202
+ options=None),
1203
+ ],
1204
+ extensions=[],
1205
+ nested_types=[],
1206
+ enum_types=[],
1207
+ options=None,
1208
+ is_extendable=False,
1209
+ syntax='proto2',
1210
+ extension_ranges=[],
1211
+ oneofs=[],
1212
+ serialized_start=1695,
1213
+ serialized_end=1894,
1214
+ )
1215
+
1216
+ _JRCLOSS = _descriptor.Descriptor(
1217
+ name='JRCLoss',
1218
+ full_name='protos.JRCLoss',
1219
+ filename=None,
1220
+ file=DESCRIPTOR,
1221
+ containing_type=None,
1222
+ fields=[
1223
+ _descriptor.FieldDescriptor(
1224
+ name='session_name',
1225
+ full_name='protos.JRCLoss.session_name',
1226
+ index=0,
1227
+ number=1,
1228
+ type=9,
1229
+ cpp_type=9,
1230
+ label=2,
1231
+ has_default_value=False,
1232
+ default_value=_b('').decode('utf-8'),
1233
+ message_type=None,
1234
+ enum_type=None,
1235
+ containing_type=None,
1236
+ is_extension=False,
1237
+ extension_scope=None,
1238
+ options=None),
1239
+ _descriptor.FieldDescriptor(
1240
+ name='alpha',
1241
+ full_name='protos.JRCLoss.alpha',
1242
+ index=1,
1243
+ number=2,
1244
+ type=2,
1245
+ cpp_type=6,
1246
+ label=1,
1247
+ has_default_value=True,
1248
+ default_value=float(0.5),
1249
+ message_type=None,
1250
+ enum_type=None,
1251
+ containing_type=None,
1252
+ is_extension=False,
1253
+ extension_scope=None,
1254
+ options=None),
1255
+ _descriptor.FieldDescriptor(
1256
+ name='same_label_loss',
1257
+ full_name='protos.JRCLoss.same_label_loss',
1258
+ index=2,
1259
+ number=3,
1260
+ type=8,
1261
+ cpp_type=7,
1262
+ label=1,
1263
+ has_default_value=True,
1264
+ default_value=True,
1265
+ message_type=None,
1266
+ enum_type=None,
1267
+ containing_type=None,
1268
+ is_extension=False,
1269
+ extension_scope=None,
1270
+ options=None),
1271
+ _descriptor.FieldDescriptor(
1272
+ name='loss_weight_strategy',
1273
+ full_name='protos.JRCLoss.loss_weight_strategy',
1274
+ index=3,
1275
+ number=4,
1276
+ type=9,
1277
+ cpp_type=9,
1278
+ label=2,
1279
+ has_default_value=True,
1280
+ default_value=_b('fixed').decode('utf-8'),
1281
+ message_type=None,
1282
+ enum_type=None,
1283
+ containing_type=None,
1284
+ is_extension=False,
1285
+ extension_scope=None,
1286
+ options=None),
1287
+ ],
1288
+ extensions=[],
1289
+ nested_types=[],
1290
+ enum_types=[],
1291
+ options=None,
1292
+ is_extendable=False,
1293
+ syntax='proto2',
1294
+ extension_ranges=[],
1295
+ oneofs=[],
1296
+ serialized_start=1896,
1297
+ serialized_end=2015,
1298
+ )
1299
+
1300
+ _LISTWISERANKLOSS = _descriptor.Descriptor(
1301
+ name='ListwiseRankLoss',
1302
+ full_name='protos.ListwiseRankLoss',
1303
+ filename=None,
1304
+ file=DESCRIPTOR,
1305
+ containing_type=None,
1306
+ fields=[
1307
+ _descriptor.FieldDescriptor(
1308
+ name='temperature',
1309
+ full_name='protos.ListwiseRankLoss.temperature',
1310
+ index=0,
1311
+ number=1,
1312
+ type=2,
1313
+ cpp_type=6,
1314
+ label=2,
1315
+ has_default_value=True,
1316
+ default_value=float(1),
1317
+ message_type=None,
1318
+ enum_type=None,
1319
+ containing_type=None,
1320
+ is_extension=False,
1321
+ extension_scope=None,
1322
+ options=None),
1323
+ _descriptor.FieldDescriptor(
1324
+ name='session_name',
1325
+ full_name='protos.ListwiseRankLoss.session_name',
1326
+ index=1,
1327
+ number=2,
1328
+ type=9,
1329
+ cpp_type=9,
1330
+ label=1,
1331
+ has_default_value=False,
1332
+ default_value=_b('').decode('utf-8'),
1333
+ message_type=None,
1334
+ enum_type=None,
1335
+ containing_type=None,
1336
+ is_extension=False,
1337
+ extension_scope=None,
1338
+ options=None),
1339
+ _descriptor.FieldDescriptor(
1340
+ name='transform_fn',
1341
+ full_name='protos.ListwiseRankLoss.transform_fn',
1342
+ index=2,
1343
+ number=3,
1344
+ type=9,
1345
+ cpp_type=9,
1346
+ label=1,
1347
+ has_default_value=False,
1348
+ default_value=_b('').decode('utf-8'),
1349
+ message_type=None,
1350
+ enum_type=None,
1351
+ containing_type=None,
1352
+ is_extension=False,
1353
+ extension_scope=None,
1354
+ options=None),
1355
+ _descriptor.FieldDescriptor(
1356
+ name='label_is_logits',
1357
+ full_name='protos.ListwiseRankLoss.label_is_logits',
1358
+ index=3,
1359
+ number=4,
1360
+ type=8,
1361
+ cpp_type=7,
1362
+ label=1,
1363
+ has_default_value=True,
1364
+ default_value=False,
1365
+ message_type=None,
1366
+ enum_type=None,
1367
+ containing_type=None,
1368
+ is_extension=False,
1369
+ extension_scope=None,
1370
+ options=None),
1371
+ _descriptor.FieldDescriptor(
1372
+ name='scale_logits',
1373
+ full_name='protos.ListwiseRankLoss.scale_logits',
1374
+ index=4,
1375
+ number=5,
1376
+ type=8,
1377
+ cpp_type=7,
1378
+ label=1,
1379
+ has_default_value=True,
1380
+ default_value=False,
1381
+ message_type=None,
1382
+ enum_type=None,
1383
+ containing_type=None,
1384
+ is_extension=False,
1385
+ extension_scope=None,
1386
+ options=None),
1387
+ ],
1388
+ extensions=[],
1389
+ nested_types=[],
1390
+ enum_types=[],
1391
+ options=None,
1392
+ is_extendable=False,
1393
+ syntax='proto2',
1394
+ extension_ranges=[],
1395
+ oneofs=[],
1396
+ serialized_start=2018,
1397
+ serialized_end=2165,
1398
+ )
1399
+
1400
+ _LISTWISEDISTILLLOSS = _descriptor.Descriptor(
1401
+ name='ListwiseDistillLoss',
1402
+ full_name='protos.ListwiseDistillLoss',
1403
+ filename=None,
1404
+ file=DESCRIPTOR,
1405
+ containing_type=None,
1406
+ fields=[
1407
+ _descriptor.FieldDescriptor(
1408
+ name='temperature',
1409
+ full_name='protos.ListwiseDistillLoss.temperature',
1410
+ index=0,
1411
+ number=1,
1412
+ type=2,
1413
+ cpp_type=6,
1414
+ label=2,
1415
+ has_default_value=True,
1416
+ default_value=float(1),
1417
+ message_type=None,
1418
+ enum_type=None,
1419
+ containing_type=None,
1420
+ is_extension=False,
1421
+ extension_scope=None,
1422
+ options=None),
1423
+ _descriptor.FieldDescriptor(
1424
+ name='session_name',
1425
+ full_name='protos.ListwiseDistillLoss.session_name',
1426
+ index=1,
1427
+ number=2,
1428
+ type=9,
1429
+ cpp_type=9,
1430
+ label=1,
1431
+ has_default_value=False,
1432
+ default_value=_b('').decode('utf-8'),
1433
+ message_type=None,
1434
+ enum_type=None,
1435
+ containing_type=None,
1436
+ is_extension=False,
1437
+ extension_scope=None,
1438
+ options=None),
1439
+ _descriptor.FieldDescriptor(
1440
+ name='transform_fn',
1441
+ full_name='protos.ListwiseDistillLoss.transform_fn',
1442
+ index=2,
1443
+ number=3,
1444
+ type=9,
1445
+ cpp_type=9,
1446
+ label=1,
1447
+ has_default_value=False,
1448
+ default_value=_b('').decode('utf-8'),
1449
+ message_type=None,
1450
+ enum_type=None,
1451
+ containing_type=None,
1452
+ is_extension=False,
1453
+ extension_scope=None,
1454
+ options=None),
1455
+ _descriptor.FieldDescriptor(
1456
+ name='label_clip_max_value',
1457
+ full_name='protos.ListwiseDistillLoss.label_clip_max_value',
1458
+ index=3,
1459
+ number=4,
1460
+ type=2,
1461
+ cpp_type=6,
1462
+ label=1,
1463
+ has_default_value=True,
1464
+ default_value=float(512),
1465
+ message_type=None,
1466
+ enum_type=None,
1467
+ containing_type=None,
1468
+ is_extension=False,
1469
+ extension_scope=None,
1470
+ options=None),
1471
+ _descriptor.FieldDescriptor(
1472
+ name='scale_logits',
1473
+ full_name='protos.ListwiseDistillLoss.scale_logits',
1474
+ index=4,
1475
+ number=5,
1476
+ type=8,
1477
+ cpp_type=7,
1478
+ label=1,
1479
+ has_default_value=True,
1480
+ default_value=False,
1481
+ message_type=None,
1482
+ enum_type=None,
1483
+ containing_type=None,
1484
+ is_extension=False,
1485
+ extension_scope=None,
1486
+ options=None),
1487
+ ],
1488
+ extensions=[],
1489
+ nested_types=[],
1490
+ enum_types=[],
1491
+ options=None,
1492
+ is_extendable=False,
1493
+ syntax='proto2',
1494
+ extension_ranges=[],
1495
+ oneofs=[],
1496
+ serialized_start=2168,
1497
+ serialized_end=2321,
1498
+ )
1499
+
1500
+ _LOSS.fields_by_name['loss_type'].enum_type = _LOSSTYPE
1501
+ _LOSS.fields_by_name['f1_reweighted_loss'].message_type = _F1REWEIGHEDLOSS
1502
+ _LOSS.fields_by_name[
1503
+ 'softmax_loss'].message_type = _SOFTMAXCROSSENTROPYWITHNEGATIVEMINING
1504
+ _LOSS.fields_by_name['circle_loss'].message_type = _CIRCLELOSS
1505
+ _LOSS.fields_by_name['multi_simi_loss'].message_type = _MULTISIMILARITYLOSS
1506
+ _LOSS.fields_by_name['binary_focal_loss'].message_type = _BINARYFOCALLOSS
1507
+ _LOSS.fields_by_name['pairwise_loss'].message_type = _PAIRWISELOSS
1508
+ _LOSS.fields_by_name['pairwise_focal_loss'].message_type = _PAIRWISEFOCALLOSS
1509
+ _LOSS.fields_by_name[
1510
+ 'pairwise_logistic_loss'].message_type = _PAIRWISELOGISTICLOSS
1511
+ _LOSS.fields_by_name['jrc_loss'].message_type = _JRCLOSS
1512
+ _LOSS.fields_by_name['pairwise_hinge_loss'].message_type = _PAIRWISEHINGELOSS
1513
+ _LOSS.fields_by_name['listwise_rank_loss'].message_type = _LISTWISERANKLOSS
1514
+ _LOSS.fields_by_name[
1515
+ 'listwise_distill_loss'].message_type = _LISTWISEDISTILLLOSS
1516
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1517
+ _LOSS.fields_by_name['f1_reweighted_loss'])
1518
+ _LOSS.fields_by_name[
1519
+ 'f1_reweighted_loss'].containing_oneof = _LOSS.oneofs_by_name['loss_param']
1520
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1521
+ _LOSS.fields_by_name['softmax_loss'])
1522
+ _LOSS.fields_by_name['softmax_loss'].containing_oneof = _LOSS.oneofs_by_name[
1523
+ 'loss_param']
1524
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1525
+ _LOSS.fields_by_name['circle_loss'])
1526
+ _LOSS.fields_by_name['circle_loss'].containing_oneof = _LOSS.oneofs_by_name[
1527
+ 'loss_param']
1528
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1529
+ _LOSS.fields_by_name['multi_simi_loss'])
1530
+ _LOSS.fields_by_name['multi_simi_loss'].containing_oneof = _LOSS.oneofs_by_name[
1531
+ 'loss_param']
1532
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1533
+ _LOSS.fields_by_name['binary_focal_loss'])
1534
+ _LOSS.fields_by_name[
1535
+ 'binary_focal_loss'].containing_oneof = _LOSS.oneofs_by_name['loss_param']
1536
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1537
+ _LOSS.fields_by_name['pairwise_loss'])
1538
+ _LOSS.fields_by_name['pairwise_loss'].containing_oneof = _LOSS.oneofs_by_name[
1539
+ 'loss_param']
1540
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1541
+ _LOSS.fields_by_name['pairwise_focal_loss'])
1542
+ _LOSS.fields_by_name[
1543
+ 'pairwise_focal_loss'].containing_oneof = _LOSS.oneofs_by_name['loss_param']
1544
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1545
+ _LOSS.fields_by_name['pairwise_logistic_loss'])
1546
+ _LOSS.fields_by_name[
1547
+ 'pairwise_logistic_loss'].containing_oneof = _LOSS.oneofs_by_name[
1548
+ 'loss_param']
1549
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1550
+ _LOSS.fields_by_name['jrc_loss'])
1551
+ _LOSS.fields_by_name['jrc_loss'].containing_oneof = _LOSS.oneofs_by_name[
1552
+ 'loss_param']
1553
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1554
+ _LOSS.fields_by_name['pairwise_hinge_loss'])
1555
+ _LOSS.fields_by_name[
1556
+ 'pairwise_hinge_loss'].containing_oneof = _LOSS.oneofs_by_name['loss_param']
1557
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1558
+ _LOSS.fields_by_name['listwise_rank_loss'])
1559
+ _LOSS.fields_by_name[
1560
+ 'listwise_rank_loss'].containing_oneof = _LOSS.oneofs_by_name['loss_param']
1561
+ _LOSS.oneofs_by_name['loss_param'].fields.append(
1562
+ _LOSS.fields_by_name['listwise_distill_loss'])
1563
+ _LOSS.fields_by_name[
1564
+ 'listwise_distill_loss'].containing_oneof = _LOSS.oneofs_by_name[
1565
+ 'loss_param']
1566
+ DESCRIPTOR.message_types_by_name['Loss'] = _LOSS
1567
+ DESCRIPTOR.message_types_by_name[
1568
+ 'SoftmaxCrossEntropyWithNegativeMining'] = _SOFTMAXCROSSENTROPYWITHNEGATIVEMINING
1569
+ DESCRIPTOR.message_types_by_name['CircleLoss'] = _CIRCLELOSS
1570
+ DESCRIPTOR.message_types_by_name['MultiSimilarityLoss'] = _MULTISIMILARITYLOSS
1571
+ DESCRIPTOR.message_types_by_name['F1ReweighedLoss'] = _F1REWEIGHEDLOSS
1572
+ DESCRIPTOR.message_types_by_name['BinaryFocalLoss'] = _BINARYFOCALLOSS
1573
+ DESCRIPTOR.message_types_by_name['PairwiseLoss'] = _PAIRWISELOSS
1574
+ DESCRIPTOR.message_types_by_name['PairwiseFocalLoss'] = _PAIRWISEFOCALLOSS
1575
+ DESCRIPTOR.message_types_by_name['PairwiseLogisticLoss'] = _PAIRWISELOGISTICLOSS
1576
+ DESCRIPTOR.message_types_by_name['PairwiseHingeLoss'] = _PAIRWISEHINGELOSS
1577
+ DESCRIPTOR.message_types_by_name['JRCLoss'] = _JRCLOSS
1578
+ DESCRIPTOR.message_types_by_name['ListwiseRankLoss'] = _LISTWISERANKLOSS
1579
+ DESCRIPTOR.message_types_by_name['ListwiseDistillLoss'] = _LISTWISEDISTILLLOSS
1580
+ DESCRIPTOR.enum_types_by_name['LossType'] = _LOSSTYPE
1581
+ _sym_db.RegisterFileDescriptor(DESCRIPTOR)
1582
+
1583
+ Loss = _reflection.GeneratedProtocolMessageType(
1584
+ 'Loss',
1585
+ (_message.Message,),
1586
+ dict(
1587
+ DESCRIPTOR=_LOSS,
1588
+ __module__='easy_rec.python.protos.loss_pb2'
1589
+ # @@protoc_insertion_point(class_scope:protos.Loss)
1590
+ ))
1591
+ _sym_db.RegisterMessage(Loss)
1592
+
1593
+ SoftmaxCrossEntropyWithNegativeMining = _reflection.GeneratedProtocolMessageType(
1594
+ 'SoftmaxCrossEntropyWithNegativeMining',
1595
+ (_message.Message,),
1596
+ dict(
1597
+ DESCRIPTOR=_SOFTMAXCROSSENTROPYWITHNEGATIVEMINING,
1598
+ __module__='easy_rec.python.protos.loss_pb2'
1599
+ # @@protoc_insertion_point(class_scope:protos.SoftmaxCrossEntropyWithNegativeMining)
1600
+ ))
1601
+ _sym_db.RegisterMessage(SoftmaxCrossEntropyWithNegativeMining)
1602
+
1603
+ CircleLoss = _reflection.GeneratedProtocolMessageType(
1604
+ 'CircleLoss',
1605
+ (_message.Message,),
1606
+ dict(
1607
+ DESCRIPTOR=_CIRCLELOSS,
1608
+ __module__='easy_rec.python.protos.loss_pb2'
1609
+ # @@protoc_insertion_point(class_scope:protos.CircleLoss)
1610
+ ))
1611
+ _sym_db.RegisterMessage(CircleLoss)
1612
+
1613
+ MultiSimilarityLoss = _reflection.GeneratedProtocolMessageType(
1614
+ 'MultiSimilarityLoss',
1615
+ (_message.Message,),
1616
+ dict(
1617
+ DESCRIPTOR=_MULTISIMILARITYLOSS,
1618
+ __module__='easy_rec.python.protos.loss_pb2'
1619
+ # @@protoc_insertion_point(class_scope:protos.MultiSimilarityLoss)
1620
+ ))
1621
+ _sym_db.RegisterMessage(MultiSimilarityLoss)
1622
+
1623
+ F1ReweighedLoss = _reflection.GeneratedProtocolMessageType(
1624
+ 'F1ReweighedLoss',
1625
+ (_message.Message,),
1626
+ dict(
1627
+ DESCRIPTOR=_F1REWEIGHEDLOSS,
1628
+ __module__='easy_rec.python.protos.loss_pb2'
1629
+ # @@protoc_insertion_point(class_scope:protos.F1ReweighedLoss)
1630
+ ))
1631
+ _sym_db.RegisterMessage(F1ReweighedLoss)
1632
+
1633
+ BinaryFocalLoss = _reflection.GeneratedProtocolMessageType(
1634
+ 'BinaryFocalLoss',
1635
+ (_message.Message,),
1636
+ dict(
1637
+ DESCRIPTOR=_BINARYFOCALLOSS,
1638
+ __module__='easy_rec.python.protos.loss_pb2'
1639
+ # @@protoc_insertion_point(class_scope:protos.BinaryFocalLoss)
1640
+ ))
1641
+ _sym_db.RegisterMessage(BinaryFocalLoss)
1642
+
1643
+ PairwiseLoss = _reflection.GeneratedProtocolMessageType(
1644
+ 'PairwiseLoss',
1645
+ (_message.Message,),
1646
+ dict(
1647
+ DESCRIPTOR=_PAIRWISELOSS,
1648
+ __module__='easy_rec.python.protos.loss_pb2'
1649
+ # @@protoc_insertion_point(class_scope:protos.PairwiseLoss)
1650
+ ))
1651
+ _sym_db.RegisterMessage(PairwiseLoss)
1652
+
1653
+ PairwiseFocalLoss = _reflection.GeneratedProtocolMessageType(
1654
+ 'PairwiseFocalLoss',
1655
+ (_message.Message,),
1656
+ dict(
1657
+ DESCRIPTOR=_PAIRWISEFOCALLOSS,
1658
+ __module__='easy_rec.python.protos.loss_pb2'
1659
+ # @@protoc_insertion_point(class_scope:protos.PairwiseFocalLoss)
1660
+ ))
1661
+ _sym_db.RegisterMessage(PairwiseFocalLoss)
1662
+
1663
+ PairwiseLogisticLoss = _reflection.GeneratedProtocolMessageType(
1664
+ 'PairwiseLogisticLoss',
1665
+ (_message.Message,),
1666
+ dict(
1667
+ DESCRIPTOR=_PAIRWISELOGISTICLOSS,
1668
+ __module__='easy_rec.python.protos.loss_pb2'
1669
+ # @@protoc_insertion_point(class_scope:protos.PairwiseLogisticLoss)
1670
+ ))
1671
+ _sym_db.RegisterMessage(PairwiseLogisticLoss)
1672
+
1673
+ PairwiseHingeLoss = _reflection.GeneratedProtocolMessageType(
1674
+ 'PairwiseHingeLoss',
1675
+ (_message.Message,),
1676
+ dict(
1677
+ DESCRIPTOR=_PAIRWISEHINGELOSS,
1678
+ __module__='easy_rec.python.protos.loss_pb2'
1679
+ # @@protoc_insertion_point(class_scope:protos.PairwiseHingeLoss)
1680
+ ))
1681
+ _sym_db.RegisterMessage(PairwiseHingeLoss)
1682
+
1683
+ JRCLoss = _reflection.GeneratedProtocolMessageType(
1684
+ 'JRCLoss',
1685
+ (_message.Message,),
1686
+ dict(
1687
+ DESCRIPTOR=_JRCLOSS,
1688
+ __module__='easy_rec.python.protos.loss_pb2'
1689
+ # @@protoc_insertion_point(class_scope:protos.JRCLoss)
1690
+ ))
1691
+ _sym_db.RegisterMessage(JRCLoss)
1692
+
1693
+ ListwiseRankLoss = _reflection.GeneratedProtocolMessageType(
1694
+ 'ListwiseRankLoss',
1695
+ (_message.Message,),
1696
+ dict(
1697
+ DESCRIPTOR=_LISTWISERANKLOSS,
1698
+ __module__='easy_rec.python.protos.loss_pb2'
1699
+ # @@protoc_insertion_point(class_scope:protos.ListwiseRankLoss)
1700
+ ))
1701
+ _sym_db.RegisterMessage(ListwiseRankLoss)
1702
+
1703
+ ListwiseDistillLoss = _reflection.GeneratedProtocolMessageType(
1704
+ 'ListwiseDistillLoss',
1705
+ (_message.Message,),
1706
+ dict(
1707
+ DESCRIPTOR=_LISTWISEDISTILLLOSS,
1708
+ __module__='easy_rec.python.protos.loss_pb2'
1709
+ # @@protoc_insertion_point(class_scope:protos.ListwiseDistillLoss)
1710
+ ))
1711
+ _sym_db.RegisterMessage(ListwiseDistillLoss)
1712
+
1713
+ # @@protoc_insertion_point(module_scope)