From 1e31401259914a858d025df73aa312ddd123c33c Mon Sep 17 00:00:00 2001 From: Aijun Bai Date: Wed, 1 Feb 2023 01:53:09 +0000 Subject: [PATCH] Update version for release 0.5.2 PiperOrigin-RevId: 506166467 --- docs/api_docs/python/tfr.md | 2 +- docs/api_docs/python/tfr/_api_cache.json | 139 +++++--- docs/api_docs/python/tfr/_redirects.yaml | 2 + docs/api_docs/python/tfr/_toc.yaml | 6 +- docs/api_docs/python/tfr/all_symbols.md | 4 +- docs/api_docs/python/tfr/api_report.pb | Bin 67006 -> 70893 bytes .../api_docs/python/tfr/extension/pipeline.md | 33 -- .../tfr/extension/pipeline/RankingPipeline.md | 6 +- docs/api_docs/python/tfr/extension/premade.md | 1 - .../tfr/extension/premade/TFRBertConfig.md | 34 +- .../extension/premade/TFRBertDataConfig.md | 36 +- .../extension/premade/TFRBertModelBuilder.md | 3 +- .../extension/premade/TFRBertModelConfig.md | 36 +- .../tfr/extension/premade/TFRBertTask.md | 19 +- .../tfr/extension/premade/tfrbert_task.md | 1 - docs/api_docs/python/tfr/extension/task.md | 1 - .../tfr/extension/task/RankingDataConfig.md | 36 +- .../python/tfr/extension/task/RankingTask.md | 25 +- .../tfr/extension/task/RankingTaskConfig.md | 34 +- .../python/tfr/keras/layers/Bilinear.md | 41 +-- .../python/tfr/keras/layers/ConcatFeatures.md | 26 +- .../layers/DocumentInteractionAttention.md | 25 +- .../python/tfr/keras/layers/FlattenList.md | 26 +- .../python/tfr/keras/layers/GAMLayer.md | 26 +- .../python/tfr/keras/layers/RestoreList.md | 26 +- .../tfr/keras/layers/SelfAttentionMask.md | 28 +- docs/api_docs/python/tfr/keras/losses.md | 3 + .../python/tfr/keras/losses/ApproxMRRLoss.md | 16 +- .../python/tfr/keras/losses/ApproxNDCGLoss.md | 16 +- .../python/tfr/keras/losses/ClickEMLoss.md | 7 +- .../tfr/keras/losses/CoupledRankDistilLoss.md | 211 ++++++++++++ .../tfr/keras/losses/DCGLambdaWeight.md | 4 +- .../tfr/keras/losses/GumbelApproxNDCGLoss.md | 15 +- .../tfr/keras/losses/LabelDiffLambdaWeight.md | 6 +- .../tfr/keras/losses/ListMLELambdaWeight.md | 4 +- .../python/tfr/keras/losses/ListMLELoss.md | 8 +- .../tfr/keras/losses/MeanSquaredLoss.md | 6 +- .../python/tfr/keras/losses/MixtureEMLoss.md | 11 +- .../tfr/keras/losses/NDCGLambdaWeight.md | 4 +- .../tfr/keras/losses/NDCGLambdaWeightV2.md | 6 +- .../python/tfr/keras/losses/OrdinalLoss.md | 16 +- .../tfr/keras/losses/PairwiseHingeLoss.md | 10 +- .../tfr/keras/losses/PairwiseLogisticLoss.md | 10 +- .../tfr/keras/losses/PairwiseMSELoss.md | 13 +- .../keras/losses/PairwiseSoftZeroOneLoss.md | 10 +- .../tfr/keras/losses/PrecisionLambdaWeight.md | 4 +- .../python/tfr/keras/losses/RankingLossKey.md | 10 +- .../keras/losses/SigmoidCrossEntropyLoss.md | 13 +- .../python/tfr/keras/losses/SoftmaxLoss.md | 8 +- .../tfr/keras/losses/UniqueSoftmaxLoss.md | 15 +- docs/api_docs/python/tfr/keras/losses/get.md | 2 +- docs/api_docs/python/tfr/keras/metrics.md | 1 - .../python/tfr/keras/metrics/ARPMetric.md | 25 +- .../tfr/keras/metrics/AlphaDCGMetric.md | 25 +- .../python/tfr/keras/metrics/DCGMetric.md | 25 +- .../python/tfr/keras/metrics/HitsMetric.md | 48 ++- .../python/tfr/keras/metrics/MRRMetric.md | 25 +- .../metrics/MeanAveragePrecisionMetric.md | 25 +- .../python/tfr/keras/metrics/NDCGMetric.md | 25 +- .../python/tfr/keras/metrics/OPAMetric.md | 25 +- .../tfr/keras/metrics/PrecisionIAMetric.md | 26 +- .../tfr/keras/metrics/PrecisionMetric.md | 25 +- .../python/tfr/keras/metrics/RecallMetric.md | 25 +- docs/api_docs/python/tfr/keras/pipeline.md | 3 - .../tfr/keras/pipeline/BaseDatasetBuilder.md | 8 +- .../tfr/keras/pipeline/DatasetHparams.md | 2 + .../pipeline/DatasetHparams/dataset_reader.md | 273 ++++++++++++--- .../pipeline/MultiLabelDatasetBuilder.md | 8 +- .../keras/pipeline/MultiObjectivePipeline.md | 312 ------------------ .../tfr/keras/pipeline/NullDatasetBuilder.md | 8 +- .../tfr/keras/pipeline/PipelineHparams.md | 2 + .../keras/pipeline/SimpleDatasetBuilder.md | 8 +- docs/api_docs/python/tfr/utils.md | 5 + docs/api_docs/python/tfr/utils/de_noise.md | 2 +- .../python/tfr/utils/gather_per_row.md | 2 +- .../python/tfr/utils/is_label_valid.md | 2 +- .../tfr/utils/organize_valid_indices.md | 2 +- .../python/tfr/utils/padded_nd_indices.md | 2 +- .../tfr/utils/parse_keys_and_weights.md | 63 ++++ .../python/tfr/utils/ragged_to_dense.md | 2 +- .../python/tfr/utils/reshape_first_ndims.md | 2 +- .../python/tfr/utils/reshape_to_2d.md | 2 +- .../python/tfr/utils/shuffle_valid_indices.md | 2 +- .../python/tfr/utils/sort_by_scores.md | 2 +- .../api_docs/python/tfr/utils/sorted_ranks.md | 2 +- tensorflow_ranking/python/version.py | 2 +- tensorflow_ranking/tools/pip_package/setup.py | 2 +- 87 files changed, 1312 insertions(+), 750 deletions(-) create mode 100644 docs/api_docs/python/tfr/keras/losses/CoupledRankDistilLoss.md delete mode 100644 docs/api_docs/python/tfr/keras/pipeline/MultiObjectivePipeline.md create mode 100644 docs/api_docs/python/tfr/utils/parse_keys_and_weights.md diff --git a/docs/api_docs/python/tfr.md b/docs/api_docs/python/tfr.md index 926920a..c3abf74 100644 --- a/docs/api_docs/python/tfr.md +++ b/docs/api_docs/python/tfr.md @@ -42,7 +42,7 @@ TensorFlow Ranking, the project to build ranking models on TensorFlow. **version** -`'0.5.0.dev'` +`'0.5.2.dev'` diff --git a/docs/api_docs/python/tfr/_api_cache.json b/docs/api_docs/python/tfr/_api_cache.json index d5a3c63..8ed2641 100644 --- a/docs/api_docs/python/tfr/_api_cache.json +++ b/docs/api_docs/python/tfr/_api_cache.json @@ -10,9 +10,6 @@ "tfr.extension.pipeline.RankingPipeline.__lt__": "tfr.keras.layers.Bilinear.__lt__", "tfr.extension.pipeline.RankingPipeline.__ne__": "tfr.keras.layers.Bilinear.__ne__", "tfr.extension.pipeline.RankingPipeline.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__", - "tfr.extension.pipeline.absolute_import": "tfr.keras.metrics.absolute_import", - "tfr.extension.pipeline.division": "tfr.keras.metrics.division", - "tfr.extension.pipeline.print_function": "tfr.keras.metrics.print_function", "tfr.extension.premade.TFRBertConfig.__ge__": "tfr.keras.layers.Bilinear.__ge__", "tfr.extension.premade.TFRBertConfig.__gt__": "tfr.keras.layers.Bilinear.__gt__", "tfr.extension.premade.TFRBertConfig.__le__": "tfr.keras.layers.Bilinear.__le__", @@ -304,12 +301,14 @@ "tfr.keras.layers.ConcatFeatures.activity_regularizer": "tfr.keras.layers.Bilinear.activity_regularizer", "tfr.keras.layers.ConcatFeatures.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.ConcatFeatures.add_metric": "tfr.keras.layers.Bilinear.add_metric", + "tfr.keras.layers.ConcatFeatures.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.ConcatFeatures.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.ConcatFeatures.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.ConcatFeatures.count_params": "tfr.keras.layers.Bilinear.count_params", "tfr.keras.layers.ConcatFeatures.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.ConcatFeatures.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.ConcatFeatures.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.ConcatFeatures.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.ConcatFeatures.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.ConcatFeatures.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.ConcatFeatures.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -337,6 +336,7 @@ "tfr.keras.layers.DocumentInteractionAttention.activity_regularizer": "tfr.keras.layers.Bilinear.activity_regularizer", "tfr.keras.layers.DocumentInteractionAttention.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.DocumentInteractionAttention.add_metric": "tfr.keras.layers.Bilinear.add_metric", + "tfr.keras.layers.DocumentInteractionAttention.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.DocumentInteractionAttention.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.DocumentInteractionAttention.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.DocumentInteractionAttention.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -344,6 +344,7 @@ "tfr.keras.layers.DocumentInteractionAttention.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.DocumentInteractionAttention.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.DocumentInteractionAttention.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.DocumentInteractionAttention.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.DocumentInteractionAttention.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.DocumentInteractionAttention.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.DocumentInteractionAttention.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -372,6 +373,7 @@ "tfr.keras.layers.FlattenList.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.FlattenList.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.layers.FlattenList.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.layers.FlattenList.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.FlattenList.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.FlattenList.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.FlattenList.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -379,6 +381,7 @@ "tfr.keras.layers.FlattenList.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.FlattenList.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.FlattenList.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.FlattenList.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.FlattenList.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.FlattenList.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.FlattenList.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -407,6 +410,7 @@ "tfr.keras.layers.GAMLayer.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.GAMLayer.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.layers.GAMLayer.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.layers.GAMLayer.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.GAMLayer.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.GAMLayer.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.GAMLayer.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -414,6 +418,7 @@ "tfr.keras.layers.GAMLayer.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.GAMLayer.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.GAMLayer.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.GAMLayer.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.GAMLayer.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.GAMLayer.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.GAMLayer.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -442,6 +447,7 @@ "tfr.keras.layers.RestoreList.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.RestoreList.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.layers.RestoreList.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.layers.RestoreList.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.RestoreList.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.RestoreList.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.RestoreList.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -449,6 +455,7 @@ "tfr.keras.layers.RestoreList.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.RestoreList.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.RestoreList.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.RestoreList.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.RestoreList.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.RestoreList.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.RestoreList.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -477,6 +484,7 @@ "tfr.keras.layers.SelfAttentionMask.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.layers.SelfAttentionMask.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.layers.SelfAttentionMask.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.layers.SelfAttentionMask.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.layers.SelfAttentionMask.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.layers.SelfAttentionMask.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.layers.SelfAttentionMask.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -484,6 +492,7 @@ "tfr.keras.layers.SelfAttentionMask.dtype": "tfr.keras.layers.Bilinear.dtype", "tfr.keras.layers.SelfAttentionMask.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.layers.SelfAttentionMask.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.layers.SelfAttentionMask.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.layers.SelfAttentionMask.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.layers.SelfAttentionMask.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.layers.SelfAttentionMask.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -523,6 +532,14 @@ "tfr.keras.losses.ClickEMLoss.__lt__": "tfr.keras.layers.Bilinear.__lt__", "tfr.keras.losses.ClickEMLoss.__ne__": "tfr.keras.layers.Bilinear.__ne__", "tfr.keras.losses.ClickEMLoss.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__", + "tfr.keras.losses.CoupledRankDistilLoss.__call__": "tfr.keras.losses.ApproxMRRLoss.__call__", + "tfr.keras.losses.CoupledRankDistilLoss.__eq__": "tfr.keras.layers.Bilinear.__eq__", + "tfr.keras.losses.CoupledRankDistilLoss.__ge__": "tfr.keras.layers.Bilinear.__ge__", + "tfr.keras.losses.CoupledRankDistilLoss.__gt__": "tfr.keras.layers.Bilinear.__gt__", + "tfr.keras.losses.CoupledRankDistilLoss.__le__": "tfr.keras.layers.Bilinear.__le__", + "tfr.keras.losses.CoupledRankDistilLoss.__lt__": "tfr.keras.layers.Bilinear.__lt__", + "tfr.keras.losses.CoupledRankDistilLoss.__ne__": "tfr.keras.layers.Bilinear.__ne__", + "tfr.keras.losses.CoupledRankDistilLoss.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__", "tfr.keras.losses.DCGLambdaWeight.__eq__": "tfr.keras.layers.Bilinear.__eq__", "tfr.keras.losses.DCGLambdaWeight.__ge__": "tfr.keras.layers.Bilinear.__ge__", "tfr.keras.losses.DCGLambdaWeight.__gt__": "tfr.keras.layers.Bilinear.__gt__", @@ -690,12 +707,14 @@ "tfr.keras.metrics.ARPMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.ARPMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.ARPMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.ARPMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.ARPMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.ARPMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.ARPMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", "tfr.keras.metrics.ARPMetric.count_params": "tfr.keras.layers.Bilinear.count_params", "tfr.keras.metrics.ARPMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.ARPMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.ARPMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.ARPMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.ARPMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.ARPMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -722,6 +741,7 @@ "tfr.keras.metrics.AlphaDCGMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.AlphaDCGMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.AlphaDCGMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.AlphaDCGMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.AlphaDCGMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.AlphaDCGMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.AlphaDCGMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -729,6 +749,7 @@ "tfr.keras.metrics.AlphaDCGMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.AlphaDCGMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.AlphaDCGMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.AlphaDCGMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.AlphaDCGMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.AlphaDCGMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.AlphaDCGMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -761,6 +782,7 @@ "tfr.keras.metrics.DCGMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.DCGMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.DCGMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.DCGMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.DCGMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.DCGMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.DCGMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -768,6 +790,7 @@ "tfr.keras.metrics.DCGMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.DCGMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.DCGMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.DCGMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.DCGMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.DCGMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.DCGMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -800,6 +823,7 @@ "tfr.keras.metrics.HitsMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.HitsMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.HitsMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.HitsMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.HitsMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.HitsMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.HitsMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -807,6 +831,7 @@ "tfr.keras.metrics.HitsMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.HitsMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.HitsMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.HitsMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.HitsMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.HitsMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.HitsMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -839,6 +864,7 @@ "tfr.keras.metrics.MRRMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.MRRMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.MRRMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.MRRMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.MRRMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.MRRMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.MRRMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -846,6 +872,7 @@ "tfr.keras.metrics.MRRMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.MRRMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.MRRMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.MRRMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.MRRMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.MRRMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.MRRMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -878,6 +905,7 @@ "tfr.keras.metrics.MeanAveragePrecisionMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.MeanAveragePrecisionMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.MeanAveragePrecisionMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.MeanAveragePrecisionMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -885,6 +913,7 @@ "tfr.keras.metrics.MeanAveragePrecisionMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.MeanAveragePrecisionMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.MeanAveragePrecisionMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.MeanAveragePrecisionMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.MeanAveragePrecisionMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.MeanAveragePrecisionMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.MeanAveragePrecisionMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -917,6 +946,7 @@ "tfr.keras.metrics.NDCGMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.NDCGMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.NDCGMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.NDCGMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.NDCGMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.NDCGMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.NDCGMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -924,6 +954,7 @@ "tfr.keras.metrics.NDCGMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.NDCGMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.NDCGMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.NDCGMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.NDCGMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.NDCGMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.NDCGMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -956,6 +987,7 @@ "tfr.keras.metrics.OPAMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.OPAMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.OPAMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.OPAMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.OPAMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.OPAMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.OPAMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -963,6 +995,7 @@ "tfr.keras.metrics.OPAMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.OPAMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.OPAMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.OPAMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.OPAMetric.get_config": "tfr.keras.metrics.ARPMetric.get_config", "tfr.keras.metrics.OPAMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.OPAMetric.input": "tfr.keras.layers.Bilinear.input", @@ -996,6 +1029,7 @@ "tfr.keras.metrics.PrecisionIAMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.PrecisionIAMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.PrecisionIAMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.PrecisionIAMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.PrecisionIAMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.PrecisionIAMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.PrecisionIAMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -1003,6 +1037,7 @@ "tfr.keras.metrics.PrecisionIAMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.PrecisionIAMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.PrecisionIAMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.PrecisionIAMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.PrecisionIAMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.PrecisionIAMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.PrecisionIAMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -1035,6 +1070,7 @@ "tfr.keras.metrics.PrecisionMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.PrecisionMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.PrecisionMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.PrecisionMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.PrecisionMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.PrecisionMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.PrecisionMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -1042,6 +1078,7 @@ "tfr.keras.metrics.PrecisionMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.PrecisionMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.PrecisionMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.PrecisionMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.PrecisionMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.PrecisionMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.PrecisionMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -1082,6 +1119,7 @@ "tfr.keras.metrics.RecallMetric.add_loss": "tfr.keras.layers.Bilinear.add_loss", "tfr.keras.metrics.RecallMetric.add_metric": "tfr.keras.layers.Bilinear.add_metric", "tfr.keras.metrics.RecallMetric.build": "tfr.keras.layers.ConcatFeatures.build", + "tfr.keras.metrics.RecallMetric.build_from_config": "tfr.keras.layers.Bilinear.build_from_config", "tfr.keras.metrics.RecallMetric.compute_dtype": "tfr.keras.layers.Bilinear.compute_dtype", "tfr.keras.metrics.RecallMetric.compute_mask": "tfr.keras.layers.Bilinear.compute_mask", "tfr.keras.metrics.RecallMetric.compute_output_shape": "tfr.keras.layers.ConcatFeatures.compute_output_shape", @@ -1089,6 +1127,7 @@ "tfr.keras.metrics.RecallMetric.dtype": "tfr.keras.metrics.ARPMetric.dtype", "tfr.keras.metrics.RecallMetric.dtype_policy": "tfr.keras.layers.Bilinear.dtype_policy", "tfr.keras.metrics.RecallMetric.dynamic": "tfr.keras.layers.Bilinear.dynamic", + "tfr.keras.metrics.RecallMetric.get_build_config": "tfr.keras.layers.Bilinear.get_build_config", "tfr.keras.metrics.RecallMetric.get_weights": "tfr.keras.layers.Bilinear.get_weights", "tfr.keras.metrics.RecallMetric.input": "tfr.keras.layers.Bilinear.input", "tfr.keras.metrics.RecallMetric.input_spec": "tfr.keras.layers.Bilinear.input_spec", @@ -1258,19 +1297,6 @@ "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_signatures": "tfr.keras.pipeline.BaseDatasetBuilder.build_signatures", "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_train_dataset": "tfr.keras.pipeline.BaseDatasetBuilder.build_train_dataset", "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_valid_dataset": "tfr.keras.pipeline.BaseDatasetBuilder.build_valid_dataset", - "tfr.keras.pipeline.MultiObjectivePipeline.__eq__": "tfr.keras.layers.Bilinear.__eq__", - "tfr.keras.pipeline.MultiObjectivePipeline.__ge__": "tfr.keras.layers.Bilinear.__ge__", - "tfr.keras.pipeline.MultiObjectivePipeline.__gt__": "tfr.keras.layers.Bilinear.__gt__", - "tfr.keras.pipeline.MultiObjectivePipeline.__init__": "tfr.keras.pipeline.ModelFitPipeline.__init__", - "tfr.keras.pipeline.MultiObjectivePipeline.__le__": "tfr.keras.layers.Bilinear.__le__", - "tfr.keras.pipeline.MultiObjectivePipeline.__lt__": "tfr.keras.layers.Bilinear.__lt__", - "tfr.keras.pipeline.MultiObjectivePipeline.__ne__": "tfr.keras.layers.Bilinear.__ne__", - "tfr.keras.pipeline.MultiObjectivePipeline.__new__": "tfr.keras.losses.ApproxMRRLoss.__new__", - "tfr.keras.pipeline.MultiObjectivePipeline.build_callbacks": "tfr.keras.pipeline.ModelFitPipeline.build_callbacks", - "tfr.keras.pipeline.MultiObjectivePipeline.build_metrics": "tfr.keras.pipeline.SimplePipeline.build_metrics", - "tfr.keras.pipeline.MultiObjectivePipeline.build_weighted_metrics": "tfr.keras.pipeline.SimplePipeline.build_weighted_metrics", - "tfr.keras.pipeline.MultiObjectivePipeline.export_saved_model": "tfr.keras.pipeline.ModelFitPipeline.export_saved_model", - "tfr.keras.pipeline.MultiObjectivePipeline.train_and_validate": "tfr.keras.pipeline.ModelFitPipeline.train_and_validate", "tfr.keras.pipeline.MultiTaskPipeline.__eq__": "tfr.keras.layers.Bilinear.__eq__", "tfr.keras.pipeline.MultiTaskPipeline.__ge__": "tfr.keras.layers.Bilinear.__ge__", "tfr.keras.pipeline.MultiTaskPipeline.__gt__": "tfr.keras.layers.Bilinear.__gt__", @@ -1336,6 +1362,7 @@ "tfr.keras.utils.PositiveFunction": "tfr.keras.utils.GainFunction", "tfr.keras.utils.RankDiscountFunction": "tfr.keras.utils.GainFunction", "tfr.keras.utils.TensorLike": "tfr.keras.model.TensorLike", + "tfr.utils.MetricFunction": "tfr.utils.LossFunction", "tfr.utils.TensorLike": "tfr.keras.model.TensorLike", "tfr.utils.TransformationFunction": "tfr.keras.utils.GainFunction" }, @@ -1373,9 +1400,6 @@ "tfr.extension.pipeline.RankingPipeline.__ne__": true, "tfr.extension.pipeline.RankingPipeline.__new__": true, "tfr.extension.pipeline.RankingPipeline.train_and_eval": true, - "tfr.extension.pipeline.absolute_import": true, - "tfr.extension.pipeline.division": true, - "tfr.extension.pipeline.print_function": true, "tfr.extension.premade": false, "tfr.extension.premade.DOCUMENT_ID": true, "tfr.extension.premade.QUERY_ID": true, @@ -1905,6 +1929,7 @@ "tfr.keras.layers.Bilinear.add_loss": true, "tfr.keras.layers.Bilinear.add_metric": true, "tfr.keras.layers.Bilinear.build": true, + "tfr.keras.layers.Bilinear.build_from_config": true, "tfr.keras.layers.Bilinear.compute_dtype": true, "tfr.keras.layers.Bilinear.compute_mask": true, "tfr.keras.layers.Bilinear.compute_output_shape": true, @@ -1913,6 +1938,7 @@ "tfr.keras.layers.Bilinear.dtype_policy": true, "tfr.keras.layers.Bilinear.dynamic": true, "tfr.keras.layers.Bilinear.from_config": true, + "tfr.keras.layers.Bilinear.get_build_config": true, "tfr.keras.layers.Bilinear.get_config": true, "tfr.keras.layers.Bilinear.get_weights": true, "tfr.keras.layers.Bilinear.input": true, @@ -1945,6 +1971,7 @@ "tfr.keras.layers.ConcatFeatures.add_loss": true, "tfr.keras.layers.ConcatFeatures.add_metric": true, "tfr.keras.layers.ConcatFeatures.build": true, + "tfr.keras.layers.ConcatFeatures.build_from_config": true, "tfr.keras.layers.ConcatFeatures.compute_dtype": true, "tfr.keras.layers.ConcatFeatures.compute_mask": true, "tfr.keras.layers.ConcatFeatures.compute_output_shape": true, @@ -1953,6 +1980,7 @@ "tfr.keras.layers.ConcatFeatures.dtype_policy": true, "tfr.keras.layers.ConcatFeatures.dynamic": true, "tfr.keras.layers.ConcatFeatures.from_config": true, + "tfr.keras.layers.ConcatFeatures.get_build_config": true, "tfr.keras.layers.ConcatFeatures.get_config": true, "tfr.keras.layers.ConcatFeatures.get_weights": true, "tfr.keras.layers.ConcatFeatures.input": true, @@ -1985,6 +2013,7 @@ "tfr.keras.layers.DocumentInteractionAttention.add_loss": true, "tfr.keras.layers.DocumentInteractionAttention.add_metric": true, "tfr.keras.layers.DocumentInteractionAttention.build": true, + "tfr.keras.layers.DocumentInteractionAttention.build_from_config": true, "tfr.keras.layers.DocumentInteractionAttention.compute_dtype": true, "tfr.keras.layers.DocumentInteractionAttention.compute_mask": true, "tfr.keras.layers.DocumentInteractionAttention.compute_output_shape": true, @@ -1993,6 +2022,7 @@ "tfr.keras.layers.DocumentInteractionAttention.dtype_policy": true, "tfr.keras.layers.DocumentInteractionAttention.dynamic": true, "tfr.keras.layers.DocumentInteractionAttention.from_config": true, + "tfr.keras.layers.DocumentInteractionAttention.get_build_config": true, "tfr.keras.layers.DocumentInteractionAttention.get_config": true, "tfr.keras.layers.DocumentInteractionAttention.get_weights": true, "tfr.keras.layers.DocumentInteractionAttention.input": true, @@ -2025,6 +2055,7 @@ "tfr.keras.layers.FlattenList.add_loss": true, "tfr.keras.layers.FlattenList.add_metric": true, "tfr.keras.layers.FlattenList.build": true, + "tfr.keras.layers.FlattenList.build_from_config": true, "tfr.keras.layers.FlattenList.compute_dtype": true, "tfr.keras.layers.FlattenList.compute_mask": true, "tfr.keras.layers.FlattenList.compute_output_shape": true, @@ -2033,6 +2064,7 @@ "tfr.keras.layers.FlattenList.dtype_policy": true, "tfr.keras.layers.FlattenList.dynamic": true, "tfr.keras.layers.FlattenList.from_config": true, + "tfr.keras.layers.FlattenList.get_build_config": true, "tfr.keras.layers.FlattenList.get_config": true, "tfr.keras.layers.FlattenList.get_weights": true, "tfr.keras.layers.FlattenList.input": true, @@ -2065,6 +2097,7 @@ "tfr.keras.layers.GAMLayer.add_loss": true, "tfr.keras.layers.GAMLayer.add_metric": true, "tfr.keras.layers.GAMLayer.build": true, + "tfr.keras.layers.GAMLayer.build_from_config": true, "tfr.keras.layers.GAMLayer.compute_dtype": true, "tfr.keras.layers.GAMLayer.compute_mask": true, "tfr.keras.layers.GAMLayer.compute_output_shape": true, @@ -2073,6 +2106,7 @@ "tfr.keras.layers.GAMLayer.dtype_policy": true, "tfr.keras.layers.GAMLayer.dynamic": true, "tfr.keras.layers.GAMLayer.from_config": true, + "tfr.keras.layers.GAMLayer.get_build_config": true, "tfr.keras.layers.GAMLayer.get_config": true, "tfr.keras.layers.GAMLayer.get_weights": true, "tfr.keras.layers.GAMLayer.input": true, @@ -2105,6 +2139,7 @@ "tfr.keras.layers.RestoreList.add_loss": true, "tfr.keras.layers.RestoreList.add_metric": true, "tfr.keras.layers.RestoreList.build": true, + "tfr.keras.layers.RestoreList.build_from_config": true, "tfr.keras.layers.RestoreList.compute_dtype": true, "tfr.keras.layers.RestoreList.compute_mask": true, "tfr.keras.layers.RestoreList.compute_output_shape": true, @@ -2113,6 +2148,7 @@ "tfr.keras.layers.RestoreList.dtype_policy": true, "tfr.keras.layers.RestoreList.dynamic": true, "tfr.keras.layers.RestoreList.from_config": true, + "tfr.keras.layers.RestoreList.get_build_config": true, "tfr.keras.layers.RestoreList.get_config": true, "tfr.keras.layers.RestoreList.get_weights": true, "tfr.keras.layers.RestoreList.input": true, @@ -2145,6 +2181,7 @@ "tfr.keras.layers.SelfAttentionMask.add_loss": true, "tfr.keras.layers.SelfAttentionMask.add_metric": true, "tfr.keras.layers.SelfAttentionMask.build": true, + "tfr.keras.layers.SelfAttentionMask.build_from_config": true, "tfr.keras.layers.SelfAttentionMask.compute_dtype": true, "tfr.keras.layers.SelfAttentionMask.compute_mask": true, "tfr.keras.layers.SelfAttentionMask.compute_output_shape": true, @@ -2153,6 +2190,7 @@ "tfr.keras.layers.SelfAttentionMask.dtype_policy": true, "tfr.keras.layers.SelfAttentionMask.dynamic": true, "tfr.keras.layers.SelfAttentionMask.from_config": true, + "tfr.keras.layers.SelfAttentionMask.get_build_config": true, "tfr.keras.layers.SelfAttentionMask.get_config": true, "tfr.keras.layers.SelfAttentionMask.get_weights": true, "tfr.keras.layers.SelfAttentionMask.input": true, @@ -2209,6 +2247,18 @@ "tfr.keras.losses.ClickEMLoss.__new__": true, "tfr.keras.losses.ClickEMLoss.from_config": true, "tfr.keras.losses.ClickEMLoss.get_config": true, + "tfr.keras.losses.CoupledRankDistilLoss": false, + "tfr.keras.losses.CoupledRankDistilLoss.__call__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__eq__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__ge__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__gt__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__init__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__le__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__lt__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__ne__": true, + "tfr.keras.losses.CoupledRankDistilLoss.__new__": true, + "tfr.keras.losses.CoupledRankDistilLoss.from_config": true, + "tfr.keras.losses.CoupledRankDistilLoss.get_config": true, "tfr.keras.losses.DCGLambdaWeight": false, "tfr.keras.losses.DCGLambdaWeight.__eq__": true, "tfr.keras.losses.DCGLambdaWeight.__ge__": true, @@ -2392,6 +2442,7 @@ "tfr.keras.losses.RankingLossKey": false, "tfr.keras.losses.RankingLossKey.APPROX_MRR_LOSS": true, "tfr.keras.losses.RankingLossKey.APPROX_NDCG_LOSS": true, + "tfr.keras.losses.RankingLossKey.COUPLED_RANKDISTIL_LOSS": true, "tfr.keras.losses.RankingLossKey.GUMBEL_APPROX_NDCG_LOSS": true, "tfr.keras.losses.RankingLossKey.LIST_MLE_LOSS": true, "tfr.keras.losses.RankingLossKey.MEAN_SQUARED_LOSS": true, @@ -2464,6 +2515,7 @@ "tfr.keras.metrics.ARPMetric.add_loss": true, "tfr.keras.metrics.ARPMetric.add_metric": true, "tfr.keras.metrics.ARPMetric.build": true, + "tfr.keras.metrics.ARPMetric.build_from_config": true, "tfr.keras.metrics.ARPMetric.compute_dtype": true, "tfr.keras.metrics.ARPMetric.compute_mask": true, "tfr.keras.metrics.ARPMetric.compute_output_shape": true, @@ -2472,6 +2524,7 @@ "tfr.keras.metrics.ARPMetric.dtype_policy": true, "tfr.keras.metrics.ARPMetric.dynamic": true, "tfr.keras.metrics.ARPMetric.from_config": true, + "tfr.keras.metrics.ARPMetric.get_build_config": true, "tfr.keras.metrics.ARPMetric.get_config": true, "tfr.keras.metrics.ARPMetric.get_weights": true, "tfr.keras.metrics.ARPMetric.input": true, @@ -2508,6 +2561,7 @@ "tfr.keras.metrics.AlphaDCGMetric.add_loss": true, "tfr.keras.metrics.AlphaDCGMetric.add_metric": true, "tfr.keras.metrics.AlphaDCGMetric.build": true, + "tfr.keras.metrics.AlphaDCGMetric.build_from_config": true, "tfr.keras.metrics.AlphaDCGMetric.compute_dtype": true, "tfr.keras.metrics.AlphaDCGMetric.compute_mask": true, "tfr.keras.metrics.AlphaDCGMetric.compute_output_shape": true, @@ -2516,6 +2570,7 @@ "tfr.keras.metrics.AlphaDCGMetric.dtype_policy": true, "tfr.keras.metrics.AlphaDCGMetric.dynamic": true, "tfr.keras.metrics.AlphaDCGMetric.from_config": true, + "tfr.keras.metrics.AlphaDCGMetric.get_build_config": true, "tfr.keras.metrics.AlphaDCGMetric.get_config": true, "tfr.keras.metrics.AlphaDCGMetric.get_weights": true, "tfr.keras.metrics.AlphaDCGMetric.input": true, @@ -2552,6 +2607,7 @@ "tfr.keras.metrics.DCGMetric.add_loss": true, "tfr.keras.metrics.DCGMetric.add_metric": true, "tfr.keras.metrics.DCGMetric.build": true, + "tfr.keras.metrics.DCGMetric.build_from_config": true, "tfr.keras.metrics.DCGMetric.compute_dtype": true, "tfr.keras.metrics.DCGMetric.compute_mask": true, "tfr.keras.metrics.DCGMetric.compute_output_shape": true, @@ -2560,6 +2616,7 @@ "tfr.keras.metrics.DCGMetric.dtype_policy": true, "tfr.keras.metrics.DCGMetric.dynamic": true, "tfr.keras.metrics.DCGMetric.from_config": true, + "tfr.keras.metrics.DCGMetric.get_build_config": true, "tfr.keras.metrics.DCGMetric.get_config": true, "tfr.keras.metrics.DCGMetric.get_weights": true, "tfr.keras.metrics.DCGMetric.input": true, @@ -2596,6 +2653,7 @@ "tfr.keras.metrics.HitsMetric.add_loss": true, "tfr.keras.metrics.HitsMetric.add_metric": true, "tfr.keras.metrics.HitsMetric.build": true, + "tfr.keras.metrics.HitsMetric.build_from_config": true, "tfr.keras.metrics.HitsMetric.compute_dtype": true, "tfr.keras.metrics.HitsMetric.compute_mask": true, "tfr.keras.metrics.HitsMetric.compute_output_shape": true, @@ -2604,6 +2662,7 @@ "tfr.keras.metrics.HitsMetric.dtype_policy": true, "tfr.keras.metrics.HitsMetric.dynamic": true, "tfr.keras.metrics.HitsMetric.from_config": true, + "tfr.keras.metrics.HitsMetric.get_build_config": true, "tfr.keras.metrics.HitsMetric.get_config": true, "tfr.keras.metrics.HitsMetric.get_weights": true, "tfr.keras.metrics.HitsMetric.input": true, @@ -2640,6 +2699,7 @@ "tfr.keras.metrics.MRRMetric.add_loss": true, "tfr.keras.metrics.MRRMetric.add_metric": true, "tfr.keras.metrics.MRRMetric.build": true, + "tfr.keras.metrics.MRRMetric.build_from_config": true, "tfr.keras.metrics.MRRMetric.compute_dtype": true, "tfr.keras.metrics.MRRMetric.compute_mask": true, "tfr.keras.metrics.MRRMetric.compute_output_shape": true, @@ -2648,6 +2708,7 @@ "tfr.keras.metrics.MRRMetric.dtype_policy": true, "tfr.keras.metrics.MRRMetric.dynamic": true, "tfr.keras.metrics.MRRMetric.from_config": true, + "tfr.keras.metrics.MRRMetric.get_build_config": true, "tfr.keras.metrics.MRRMetric.get_config": true, "tfr.keras.metrics.MRRMetric.get_weights": true, "tfr.keras.metrics.MRRMetric.input": true, @@ -2684,6 +2745,7 @@ "tfr.keras.metrics.MeanAveragePrecisionMetric.add_loss": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.add_metric": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.build": true, + "tfr.keras.metrics.MeanAveragePrecisionMetric.build_from_config": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_dtype": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_mask": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.compute_output_shape": true, @@ -2692,6 +2754,7 @@ "tfr.keras.metrics.MeanAveragePrecisionMetric.dtype_policy": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.dynamic": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.from_config": true, + "tfr.keras.metrics.MeanAveragePrecisionMetric.get_build_config": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.get_config": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.get_weights": true, "tfr.keras.metrics.MeanAveragePrecisionMetric.input": true, @@ -2728,6 +2791,7 @@ "tfr.keras.metrics.NDCGMetric.add_loss": true, "tfr.keras.metrics.NDCGMetric.add_metric": true, "tfr.keras.metrics.NDCGMetric.build": true, + "tfr.keras.metrics.NDCGMetric.build_from_config": true, "tfr.keras.metrics.NDCGMetric.compute_dtype": true, "tfr.keras.metrics.NDCGMetric.compute_mask": true, "tfr.keras.metrics.NDCGMetric.compute_output_shape": true, @@ -2736,6 +2800,7 @@ "tfr.keras.metrics.NDCGMetric.dtype_policy": true, "tfr.keras.metrics.NDCGMetric.dynamic": true, "tfr.keras.metrics.NDCGMetric.from_config": true, + "tfr.keras.metrics.NDCGMetric.get_build_config": true, "tfr.keras.metrics.NDCGMetric.get_config": true, "tfr.keras.metrics.NDCGMetric.get_weights": true, "tfr.keras.metrics.NDCGMetric.input": true, @@ -2772,6 +2837,7 @@ "tfr.keras.metrics.OPAMetric.add_loss": true, "tfr.keras.metrics.OPAMetric.add_metric": true, "tfr.keras.metrics.OPAMetric.build": true, + "tfr.keras.metrics.OPAMetric.build_from_config": true, "tfr.keras.metrics.OPAMetric.compute_dtype": true, "tfr.keras.metrics.OPAMetric.compute_mask": true, "tfr.keras.metrics.OPAMetric.compute_output_shape": true, @@ -2780,6 +2846,7 @@ "tfr.keras.metrics.OPAMetric.dtype_policy": true, "tfr.keras.metrics.OPAMetric.dynamic": true, "tfr.keras.metrics.OPAMetric.from_config": true, + "tfr.keras.metrics.OPAMetric.get_build_config": true, "tfr.keras.metrics.OPAMetric.get_config": true, "tfr.keras.metrics.OPAMetric.get_weights": true, "tfr.keras.metrics.OPAMetric.input": true, @@ -2816,6 +2883,7 @@ "tfr.keras.metrics.PrecisionIAMetric.add_loss": true, "tfr.keras.metrics.PrecisionIAMetric.add_metric": true, "tfr.keras.metrics.PrecisionIAMetric.build": true, + "tfr.keras.metrics.PrecisionIAMetric.build_from_config": true, "tfr.keras.metrics.PrecisionIAMetric.compute_dtype": true, "tfr.keras.metrics.PrecisionIAMetric.compute_mask": true, "tfr.keras.metrics.PrecisionIAMetric.compute_output_shape": true, @@ -2824,6 +2892,7 @@ "tfr.keras.metrics.PrecisionIAMetric.dtype_policy": true, "tfr.keras.metrics.PrecisionIAMetric.dynamic": true, "tfr.keras.metrics.PrecisionIAMetric.from_config": true, + "tfr.keras.metrics.PrecisionIAMetric.get_build_config": true, "tfr.keras.metrics.PrecisionIAMetric.get_config": true, "tfr.keras.metrics.PrecisionIAMetric.get_weights": true, "tfr.keras.metrics.PrecisionIAMetric.input": true, @@ -2860,6 +2929,7 @@ "tfr.keras.metrics.PrecisionMetric.add_loss": true, "tfr.keras.metrics.PrecisionMetric.add_metric": true, "tfr.keras.metrics.PrecisionMetric.build": true, + "tfr.keras.metrics.PrecisionMetric.build_from_config": true, "tfr.keras.metrics.PrecisionMetric.compute_dtype": true, "tfr.keras.metrics.PrecisionMetric.compute_mask": true, "tfr.keras.metrics.PrecisionMetric.compute_output_shape": true, @@ -2868,6 +2938,7 @@ "tfr.keras.metrics.PrecisionMetric.dtype_policy": true, "tfr.keras.metrics.PrecisionMetric.dynamic": true, "tfr.keras.metrics.PrecisionMetric.from_config": true, + "tfr.keras.metrics.PrecisionMetric.get_build_config": true, "tfr.keras.metrics.PrecisionMetric.get_config": true, "tfr.keras.metrics.PrecisionMetric.get_weights": true, "tfr.keras.metrics.PrecisionMetric.input": true, @@ -2923,6 +2994,7 @@ "tfr.keras.metrics.RecallMetric.add_loss": true, "tfr.keras.metrics.RecallMetric.add_metric": true, "tfr.keras.metrics.RecallMetric.build": true, + "tfr.keras.metrics.RecallMetric.build_from_config": true, "tfr.keras.metrics.RecallMetric.compute_dtype": true, "tfr.keras.metrics.RecallMetric.compute_mask": true, "tfr.keras.metrics.RecallMetric.compute_output_shape": true, @@ -2931,6 +3003,7 @@ "tfr.keras.metrics.RecallMetric.dtype_policy": true, "tfr.keras.metrics.RecallMetric.dynamic": true, "tfr.keras.metrics.RecallMetric.from_config": true, + "tfr.keras.metrics.RecallMetric.get_build_config": true, "tfr.keras.metrics.RecallMetric.get_config": true, "tfr.keras.metrics.RecallMetric.get_weights": true, "tfr.keras.metrics.RecallMetric.input": true, @@ -3175,6 +3248,7 @@ "tfr.keras.pipeline.DatasetHparams.dataset_reader.options": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.padded_batch": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.prefetch": true, + "tfr.keras.pipeline.DatasetHparams.dataset_reader.ragged_batch": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.random": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.range": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.rebatch": true, @@ -3188,6 +3262,7 @@ "tfr.keras.pipeline.DatasetHparams.dataset_reader.shuffle": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.skip": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.snapshot": true, + "tfr.keras.pipeline.DatasetHparams.dataset_reader.sparse_batch": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.take": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.take_while": true, "tfr.keras.pipeline.DatasetHparams.dataset_reader.unbatch": true, @@ -3224,21 +3299,6 @@ "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_signatures": true, "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_train_dataset": true, "tfr.keras.pipeline.MultiLabelDatasetBuilder.build_valid_dataset": true, - "tfr.keras.pipeline.MultiObjectivePipeline": false, - "tfr.keras.pipeline.MultiObjectivePipeline.__eq__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__ge__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__gt__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__init__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__le__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__lt__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__ne__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.__new__": true, - "tfr.keras.pipeline.MultiObjectivePipeline.build_callbacks": true, - "tfr.keras.pipeline.MultiObjectivePipeline.build_loss": true, - "tfr.keras.pipeline.MultiObjectivePipeline.build_metrics": true, - "tfr.keras.pipeline.MultiObjectivePipeline.build_weighted_metrics": true, - "tfr.keras.pipeline.MultiObjectivePipeline.export_saved_model": true, - "tfr.keras.pipeline.MultiObjectivePipeline.train_and_validate": true, "tfr.keras.pipeline.MultiTaskPipeline": false, "tfr.keras.pipeline.MultiTaskPipeline.__eq__": true, "tfr.keras.pipeline.MultiTaskPipeline.__ge__": true, @@ -3370,6 +3430,7 @@ "tfr.keras.utils.symmetric_log1p": false, "tfr.utils": false, "tfr.utils.LossFunction": false, + "tfr.utils.MetricFunction": false, "tfr.utils.TensorLike": false, "tfr.utils.TransformationFunction": false, "tfr.utils.de_noise": false, @@ -3377,6 +3438,7 @@ "tfr.utils.is_label_valid": false, "tfr.utils.organize_valid_indices": false, "tfr.utils.padded_nd_indices": false, + "tfr.utils.parse_keys_and_weights": false, "tfr.utils.ragged_to_dense": false, "tfr.utils.reshape_first_ndims": false, "tfr.utils.reshape_to_2d": false, @@ -3384,8 +3446,9 @@ "tfr.utils.sort_by_scores": false, "tfr.utils.sorted_ranks": false }, - "py_module_names": [ - "tfr" - ], - "site_link": null + "link_prefix": null, + "site_link": null, + "py_module_names": { + "tfr": "tensorflow_ranking" + } } diff --git a/docs/api_docs/python/tfr/_redirects.yaml b/docs/api_docs/python/tfr/_redirects.yaml index 6058818..25b45a4 100644 --- a/docs/api_docs/python/tfr/_redirects.yaml +++ b/docs/api_docs/python/tfr/_redirects.yaml @@ -25,6 +25,8 @@ redirects: to: /ranking/api_docs/python/tfr/keras/utils/GainFunction - from: /ranking/api_docs/python/tfr/keras/utils/TensorLike to: /ranking/api_docs/python/tfr/keras/model/TensorLike +- from: /ranking/api_docs/python/tfr/utils/MetricFunction + to: /ranking/api_docs/python/tfr/utils/LossFunction - from: /ranking/api_docs/python/tfr/utils/TensorLike to: /ranking/api_docs/python/tfr/keras/model/TensorLike - from: /ranking/api_docs/python/tfr/utils/TransformationFunction diff --git a/docs/api_docs/python/tfr/_toc.yaml b/docs/api_docs/python/tfr/_toc.yaml index d9f1d6a..3505887 100644 --- a/docs/api_docs/python/tfr/_toc.yaml +++ b/docs/api_docs/python/tfr/_toc.yaml @@ -113,6 +113,8 @@ toc: path: /ranking/api_docs/python/tfr/keras/losses/ApproxNDCGLoss - title: ClickEMLoss path: /ranking/api_docs/python/tfr/keras/losses/ClickEMLoss + - title: CoupledRankDistilLoss + path: /ranking/api_docs/python/tfr/keras/losses/CoupledRankDistilLoss - title: DCGLambdaWeight path: /ranking/api_docs/python/tfr/keras/losses/DCGLambdaWeight - title: GumbelApproxNDCGLoss @@ -237,8 +239,6 @@ toc: path: /ranking/api_docs/python/tfr/keras/pipeline/ModelFitPipeline - title: MultiLabelDatasetBuilder path: /ranking/api_docs/python/tfr/keras/pipeline/MultiLabelDatasetBuilder - - title: MultiObjectivePipeline - path: /ranking/api_docs/python/tfr/keras/pipeline/MultiObjectivePipeline - title: MultiTaskPipeline path: /ranking/api_docs/python/tfr/keras/pipeline/MultiTaskPipeline - title: NullDatasetBuilder @@ -301,6 +301,8 @@ toc: path: /ranking/api_docs/python/tfr/utils/organize_valid_indices - title: padded_nd_indices path: /ranking/api_docs/python/tfr/utils/padded_nd_indices + - title: parse_keys_and_weights + path: /ranking/api_docs/python/tfr/utils/parse_keys_and_weights - title: ragged_to_dense path: /ranking/api_docs/python/tfr/utils/ragged_to_dense - title: reshape_first_ndims diff --git a/docs/api_docs/python/tfr/all_symbols.md b/docs/api_docs/python/tfr/all_symbols.md index 6b1603b..ae880dc 100644 --- a/docs/api_docs/python/tfr/all_symbols.md +++ b/docs/api_docs/python/tfr/all_symbols.md @@ -61,6 +61,7 @@ * tfr.keras.losses.ApproxMRRLoss * tfr.keras.losses.ApproxNDCGLoss * tfr.keras.losses.ClickEMLoss +* tfr.keras.losses.CoupledRankDistilLoss * tfr.keras.losses.DCGLambdaWeight * tfr.keras.losses.GumbelApproxNDCGLoss * tfr.keras.losses.LabelDiffLambdaWeight @@ -120,7 +121,6 @@ * tfr.keras.pipeline.DatasetHparams.dataset_reader * tfr.keras.pipeline.ModelFitPipeline * tfr.keras.pipeline.MultiLabelDatasetBuilder -* tfr.keras.pipeline.MultiObjectivePipeline * tfr.keras.pipeline.MultiTaskPipeline * tfr.keras.pipeline.NullDatasetBuilder * tfr.keras.pipeline.PipelineHparams @@ -146,6 +146,7 @@ * tfr.keras.utils.symmetric_log1p * tfr.utils * tfr.utils.LossFunction +* tfr.utils.MetricFunction * tfr.utils.TensorLike * tfr.utils.TransformationFunction * tfr.utils.de_noise @@ -153,6 +154,7 @@ * tfr.utils.is_label_valid * tfr.utils.organize_valid_indices * tfr.utils.padded_nd_indices +* tfr.utils.parse_keys_and_weights * tfr.utils.ragged_to_dense * tfr.utils.reshape_first_ndims * tfr.utils.reshape_to_2d diff --git a/docs/api_docs/python/tfr/api_report.pb b/docs/api_docs/python/tfr/api_report.pb index e6611e2214dc970d9cd21e2e2b64039ea1668955..c8e1ebe4f3ab8143da4471da37a8a704a4bf2ca2 100644 GIT binary patch delta 2711 zcma)8dr*{h6y0UPeX!rMtf1S<@(`4V@md5XZUQ$7OX zl$(lzz-+0KsgDeKUgN;$gZ-?y#;UP!pzyxgdOH%s8fd4b8M(MKd)TRhU3EW zDY%~-LUJ^j4s62NH!+mt1kyHBuwD|5_$Y7tiBa0n6ruOhEGWw=xvQ&cxjL%g8r#s#zkzYm#^dW`pS5KNK?N z_>ua!N|q>maU4o@Qlx5kn3Z8qABFFAlP4ezO?G2YWJcmXPFZF*1DWb%-L@+_Q`^feI3ueCBEO zbm;iRxT15zlSNXZOj%^{W!KVfGl`&BG8@a%eC@Z26s2C}NUHbIoljFg)1oD551bHN zPALmZ42C>INK#2fX<=S&T6XdJxdn#uf4e8SbN$A^LlYoJ{UT8^|PPGT& zzw7~hw3Mx~V?_{-Ww^u%-wl5B0 zdJ6T8A{r|tVHLYMp`S)k(7rCFp_H{)ycJHNw~h&yKy+{c-CYV^n-t{L%6b8JUjtEn zmws1u*jxrWzm!336r!ZY#eN~=e$^$8{8$D(hz2oAYh*U;>{&$jOu;|17qc#WY zsMPe;3q)W#g}!zb)NOB@hg(o0iNt89pM9Zt{4a&lF4ggmyEt!WjZ%2nsT!}B^tLBk zxPIB{j)qU$i6shWooeJd-GIH^nTmWsn48cQ@M0w*LAN|8JqWG>(z#M~_ z*j0bhi|M*2Zr)mo%XH+{8~)O=?EL(^TzyV<`RcXU5U3RLm|ed(3HOg6G<9E(mEB`od% zOw5PSI7!91pacZj4|M_Uzzj5k(8ZYj1woS$C(mRtTSSMwM{Avb-S6)Ed4BJ5%RN$8 zvouzvR%|ScUy;Q980D`&~kvM7b^!I-Sf5Rb+dinrtr=Izu?7 zHEL3)mdPcrsbW9{#ux-`JL4sEAv`{nJTgEnsZm9t@#!G+htYaAz(ZdEQ8r%Rz6EYs z2!gl9piaFHCE*D;FW<{ut2KK`Nw{4lD1_lroRU-UuRtQ=Z%f6Zm+?z>EJDrWu+VdP7?rCm7oX?!_ z(b6&>V0J-^3~Ln1ck_fDyX$$vWH-$X<{2reVcrl(KM6j;MMk*y7+DWoE;17Lo)n(8 z_NEqP2V|b1hJ2V~^?8MwjTqQRBPOqjmmt>t13aZFnH?$7l)MFYCJ{gI_vG%gQxxCO zinZrD1X^&>h|h_X9eB~_koP?eNAZ(TA8HdquDEDSr_13iHjtMNf5y|=CTKjBBPY({ zxlM2$HKMRwhaZn)$O`D-RtHj#d7X>ZGVS?wS!CBAlshm9%XZyWVEG& zQwP%!$im3#g}jSyVoX(Xdv*o(d2#Eo{ja?#w%oyo%vyFiTv>pI`5 zf^@O~x)1za=Q}mmJN-;OD3a(Dd-FQ1Vv0uor;%v6dPN93P@#pjCldB+{O416?F`?w zp^}H@P2`fvZj+$F1`qV~6+$DU-LutRHD>!bI{Nv0GTRr!$q;+L+-uJTUmfCu&orCB zo~s~Q@Lx2Ze@@!^YiK4Fq$iheu>!!!dtq{)xCE4YQ z^0^-Whz07=_wBm^_2yeNPTuzH-}BVlyWrLfeDBs{G)BMRh9qy|giq$qunF!-+Ozc* zL1x*#6|O11cYQ65%rOOvAg-zHywo? - - - # Module: tfr.extension.pipeline @@ -32,33 +29,3 @@ derive from this class and further tailor for their needs. [`class RankingPipeline`](../../tfr/extension/pipeline/RankingPipeline.md): Class to set up the input, train and eval processes for a TF Ranking model. - - - - - - - - - - - - - - - - - -

Other Members

-absolute_import - -Instance of `__future__._Feature` -
-division - -Instance of `__future__._Feature` -
-print_function - -Instance of `__future__._Feature` -
diff --git a/docs/api_docs/python/tfr/extension/pipeline/RankingPipeline.md b/docs/api_docs/python/tfr/extension/pipeline/RankingPipeline.md index 1b93de3..72ae5c9 100644 --- a/docs/api_docs/python/tfr/extension/pipeline/RankingPipeline.md +++ b/docs/api_docs/python/tfr/extension/pipeline/RankingPipeline.md @@ -14,7 +14,7 @@ Ranking model. - diff --git a/docs/api_docs/python/tfr/keras/layers/DocumentInteractionAttention.md b/docs/api_docs/python/tfr/keras/layers/DocumentInteractionAttention.md index d7bc0a0..10ea7d4 100644 --- a/docs/api_docs/python/tfr/keras/layers/DocumentInteractionAttention.md +++ b/docs/api_docs/python/tfr/keras/layers/DocumentInteractionAttention.md @@ -9,10 +9,12 @@ description: Cross Document Interaction Attention layer. + + @@ -78,7 +80,6 @@ dia_output = dia_layer( ``` -
Note
@@ -353,7 +354,11 @@ class MyLayer(tf.keras.layers.Layer): return inputs ``` -This method can also be called directly on a Functional Model during +The same code works in distributed training: the input to `add_loss()` is +treated like a regularization loss and averaged across replicas by the training +loop (both built-in `Model.fit()` and compliant custom training loops). + +The `add_loss` method can also be called directly on a Functional Model during construction. In this case, any loss Tensors passed to this Model must be symbolic and be able to be traced back to the model's `Input`s. These losses become part of the model's topology and are tracked in `get_config`. @@ -526,6 +531,14 @@ correspond to `inputs` argument of call.

References

+

build_from_config

+ +
+build_from_config(
+    config
+)
+
+

compute_mask

@@ -605,7 +618,6 @@ instead of an integer.
 
 
 
-
  
@@ -697,6 +709,12 @@ A layer instance.
 
 

 
 
Returns

 
+
get_build_config

+
+
+get_build_config()
+

+
 
get_config

 
 View
@@ -960,7 +978,6 @@ Output tensor(s).
 
 
 
-
  
diff --git a/docs/api_docs/python/tfr/keras/layers/FlattenList.md b/docs/api_docs/python/tfr/keras/layers/FlattenList.md
index 22db2b9..50f9ec2 100644
--- a/docs/api_docs/python/tfr/keras/layers/FlattenList.md
+++ b/docs/api_docs/python/tfr/keras/layers/FlattenList.md
@@ -9,10 +9,12 @@ description: Layer to flatten the example list.
 
+
+
@@ -301,7 +303,11 @@ class MyLayer(tf.keras.layers.Layer):
     return inputs
 ```
 
-This method can also be called directly on a Functional Model during
+The same code works in distributed training: the input to `add_loss()` is
+treated like a regularization loss and averaged across replicas by the training
+loop (both built-in `Model.fit()` and compliant custom training loops).
+
+The `add_loss` method can also be called directly on a Functional Model during
 construction. In this case, any loss Tensors passed to this Model must be
 symbolic and be able to be traced back to the model's `Input`s. These losses
 become part of the model's topology and are tracked in `get_config`.
@@ -449,7 +455,7 @@ default using a `keras.Metric.Mean`.
 )
 
 
-Creates the variables of the layer (optional, for subclass implementers).
+Creates the variables of the layer (for subclass implementers).
 
 This is a method that implementers of subclasses of `Layer` or `Model` can
 override if they need a state-creation step in-between layer instantiation and
@@ -475,6 +481,14 @@ Instance of `TensorShape`, or list of instances of
 

 
 
Note

 
 
 
 
 
 
 
 
 
 

 
+
build_from_config

+
+
+build_from_config(
+    config
+)
+

+
 
compute_mask

 
 
@@ -554,7 +568,6 @@ instead of an integer.
 
 
 
-
  
@@ -646,6 +659,12 @@ A layer instance.
 
 

 
 
Returns

 
+
get_build_config

+
+
+get_build_config()
+

+
 
get_config

 
 View
@@ -909,7 +928,6 @@ Output tensor(s).
 
 
 
-
  
diff --git a/docs/api_docs/python/tfr/keras/layers/GAMLayer.md b/docs/api_docs/python/tfr/keras/layers/GAMLayer.md
index 1a7e8ed..ea947d4 100644
--- a/docs/api_docs/python/tfr/keras/layers/GAMLayer.md
+++ b/docs/api_docs/python/tfr/keras/layers/GAMLayer.md
@@ -9,10 +9,12 @@ description: Defines a generalized additive model (GAM) layer.
 
+
+
@@ -395,7 +397,11 @@ class MyLayer(tf.keras.layers.Layer):
     return inputs
 ```
 
-This method can also be called directly on a Functional Model during
+The same code works in distributed training: the input to `add_loss()` is
+treated like a regularization loss and averaged across replicas by the training
+loop (both built-in `Model.fit()` and compliant custom training loops).
+
+The `add_loss` method can also be called directly on a Functional Model during
 construction. In this case, any loss Tensors passed to this Model must be
 symbolic and be able to be traced back to the model's `Input`s. These losses
 become part of the model's topology and are tracked in `get_config`.
@@ -543,7 +549,7 @@ default using a `keras.Metric.Mean`.
 )
 
 
-Creates the variables of the layer (optional, for subclass implementers).
+Creates the variables of the layer (for subclass implementers).
 
 This is a method that implementers of subclasses of `Layer` or `Model` can
 override if they need a state-creation step in-between layer instantiation and
@@ -569,6 +575,14 @@ Instance of `TensorShape`, or list of instances of
 

 
 
Note

 
 
 
 
 
 
 
 
 
 

 
+
build_from_config

+
+
+build_from_config(
+    config
+)
+

+
 
compute_mask

 
 
@@ -648,7 +662,6 @@ instead of an integer.
 
 
 
-
  
@@ -740,6 +753,12 @@ A layer instance.
 
 

 
 
Returns

 
+
get_build_config

+
+
+get_build_config()
+

+
 
get_config

 
 View
@@ -1003,7 +1022,6 @@ Output tensor(s).
 
 
 
-
  
diff --git a/docs/api_docs/python/tfr/keras/layers/RestoreList.md b/docs/api_docs/python/tfr/keras/layers/RestoreList.md
index d8cfecc..1ae3e73 100644
--- a/docs/api_docs/python/tfr/keras/layers/RestoreList.md
+++ b/docs/api_docs/python/tfr/keras/layers/RestoreList.md
@@ -9,10 +9,12 @@ description: Output layer to restore listwise output shape.
 
+
+
@@ -278,7 +280,11 @@ class MyLayer(tf.keras.layers.Layer):
     return inputs
 ```
 
-This method can also be called directly on a Functional Model during
+The same code works in distributed training: the input to `add_loss()` is
+treated like a regularization loss and averaged across replicas by the training
+loop (both built-in `Model.fit()` and compliant custom training loops).
+
+The `add_loss` method can also be called directly on a Functional Model during
 construction. In this case, any loss Tensors passed to this Model must be
 symbolic and be able to be traced back to the model's `Input`s. These losses
 become part of the model's topology and are tracked in `get_config`.
@@ -426,7 +432,7 @@ default using a `keras.Metric.Mean`.
 )
 
 
-Creates the variables of the layer (optional, for subclass implementers).
+Creates the variables of the layer (for subclass implementers).
 
 This is a method that implementers of subclasses of `Layer` or `Model` can
 override if they need a state-creation step in-between layer instantiation and
@@ -452,6 +458,14 @@ Instance of `TensorShape`, or list of instances of
 

 
 
Note

 
 
 
 
 
 
 
 
 
 

 
+
build_from_config

+
+
+build_from_config(
+    config
+)
+

+
 
compute_mask

 
 
@@ -531,7 +545,6 @@ instead of an integer.
 
 
 
-
  
@@ -623,6 +636,12 @@ A layer instance.
 
 

 
 
Returns

 
+
get_build_config

+
+
+get_build_config()
+

+
 
get_config

 
 View
@@ -886,7 +905,6 @@ Output tensor(s).
 
 
 
-
  
diff --git a/docs/api_docs/python/tfr/keras/layers/SelfAttentionMask.md b/docs/api_docs/python/tfr/keras/layers/SelfAttentionMask.md
index de7fe0b..cc4d627 100644
--- a/docs/api_docs/python/tfr/keras/layers/SelfAttentionMask.md
+++ b/docs/api_docs/python/tfr/keras/layers/SelfAttentionMask.md
@@ -9,10 +9,12 @@ description: Create 3D attention mask from a 2D tensor mask.
 
+
+
@@ -259,7 +261,11 @@ class MyLayer(tf.keras.layers.Layer):
     return inputs
 ```
 
-This method can also be called directly on a Functional Model during
+The same code works in distributed training: the input to `add_loss()` is
+treated like a regularization loss and averaged across replicas by the training
+loop (both built-in `Model.fit()` and compliant custom training loops).
+
+The `add_loss` method can also be called directly on a Functional Model during
 construction. In this case, any loss Tensors passed to this Model must be
 symbolic and be able to be traced back to the model's `Input`s. These losses
 become part of the model's topology and are tracked in `get_config`.
@@ -407,7 +413,7 @@ default using a `keras.Metric.Mean`.
 )
 
 
-Creates the variables of the layer (optional, for subclass implementers).
+Creates the variables of the layer (for subclass implementers).
 
 This is a method that implementers of subclasses of `Layer` or `Model` can
 override if they need a state-creation step in-between layer instantiation and
@@ -433,6 +439,14 @@ Instance of `TensorShape`, or list of instances of
 

 
 
Note

 
 
 
 
 
 
 
 
 
 

 
+
build_from_config

+
+
+build_from_config(
+    config
+)
+

+
 
compute_mask

 
 
@@ -512,7 +526,6 @@ instead of an integer.
 
 
 
-
  
@@ -593,7 +606,6 @@ output of get_config.
 

 
 
Returns

 
 
-
  
@@ -605,6 +617,12 @@ A layer instance.
 
 

 
 
Returns

 
+
get_build_config

+
+
+get_build_config()
+

+
 
get_config

 
 
@@ -865,7 +883,6 @@ Output tensor(s).
 
 
 
-
  
@@ -888,7 +905,6 @@ Output tensor(s).
 

 
 
Note

 
 
-
  
diff --git a/docs/api_docs/python/tfr/keras/losses.md b/docs/api_docs/python/tfr/keras/losses.md
index af78d8f..5143f3c 100644
--- a/docs/api_docs/python/tfr/keras/losses.md
+++ b/docs/api_docs/python/tfr/keras/losses.md
@@ -31,6 +31,9 @@ approximate NDCG loss between `y_true` and `y_pred`.
 [`class ClickEMLoss`](../../tfr/keras/losses/ClickEMLoss.md): Computes click EM
 loss between `y_true` and `y_pred`.
 
+[`class CoupledRankDistilLoss`](../../tfr/keras/losses/CoupledRankDistilLoss.md):
+Computes the Rank Distil loss between `y_true` and `y_pred`.
+
 [`class DCGLambdaWeight`](../../tfr/keras/losses/DCGLambdaWeight.md): Keras
 serializable class for DCG.
 
diff --git a/docs/api_docs/python/tfr/keras/losses/ApproxMRRLoss.md b/docs/api_docs/python/tfr/keras/losses/ApproxMRRLoss.md
index 7a31c40..309f213 100644
--- a/docs/api_docs/python/tfr/keras/losses/ApproxMRRLoss.md
+++ b/docs/api_docs/python/tfr/keras/losses/ApproxMRRLoss.md
@@ -15,7 +15,7 @@ description: Computes approximate MRR loss between y_true and y_pred.
 
 

 
 
Raises