Primary symbols
tfr
tfr.data
tfr.data.build_ranking_dataset
tfr.data.build_ranking_dataset_with_parsing_fn
tfr.data.build_ranking_serving_input_receiver_fn
tfr.data.build_ranking_serving_input_receiver_fn_with_parsing_fn
tfr.data.build_sequence_example_serving_input_receiver_fn
tfr.data.build_tf_example_serving_input_receiver_fn
tfr.data.make_parsing_fn
tfr.data.parse_from_example_in_example
tfr.data.parse_from_example_list
tfr.data.parse_from_sequence_example
tfr.data.parse_from_tf_example
tfr.data.read_batched_sequence_example_dataset
tfr.extension
tfr.extension.pipeline
tfr.extension.pipeline.RankingPipeline
tfr.extension.premade
tfr.extension.premade.TFRBertConfig
tfr.extension.premade.TFRBertDataConfig
tfr.extension.premade.TFRBertDataLoader
tfr.extension.premade.TFRBertModelBuilder
tfr.extension.premade.TFRBertModelConfig
tfr.extension.premade.TFRBertScorer
tfr.extension.premade.TFRBertTask
tfr.extension.premade.TensorDict
tfr.extension.premade.TensorLike
tfr.extension.premade.tfrbert_task
tfr.extension.premade.tfrbert_task.TFRBertConfig
tfr.extension.premade.tfrbert_task.TFRBertDataConfig
tfr.extension.premade.tfrbert_task.TFRBertDataLoader
tfr.extension.premade.tfrbert_task.TFRBertModelBuilder
tfr.extension.premade.tfrbert_task.TFRBertModelConfig
tfr.extension.premade.tfrbert_task.TFRBertScorer
tfr.extension.premade.tfrbert_task.TFRBertTask
tfr.extension.premade.tfrbert_task.TensorDict
tfr.extension.premade.tfrbert_task.TensorLike
tfr.extension.task
tfr.extension.task.FeatureSpec
tfr.extension.task.RankingDataConfig
tfr.extension.task.RankingDataLoader
tfr.extension.task.RankingTask
tfr.extension.task.RankingTaskConfig
tfr.keras
tfr.keras.layers
tfr.keras.layers.Bilinear
tfr.keras.layers.ConcatFeatures
tfr.keras.layers.DocumentInteractionAttention
tfr.keras.layers.FlattenList
tfr.keras.layers.GAMLayer
tfr.keras.layers.RestoreList
tfr.keras.layers.SelfAttentionMask
tfr.keras.layers.create_tower
tfr.keras.losses
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
tfr.keras.losses.ListMLELambdaWeight
tfr.keras.losses.ListMLELoss
tfr.keras.losses.MeanSquaredLoss
tfr.keras.losses.MixtureEMLoss
tfr.keras.losses.NDCGLambdaWeight
tfr.keras.losses.NDCGLambdaWeightV2
tfr.keras.losses.OrdinalLoss
tfr.keras.losses.PairwiseHingeLoss
tfr.keras.losses.PairwiseLogisticLoss
tfr.keras.losses.PairwiseMSELoss
tfr.keras.losses.PairwiseSoftZeroOneLoss
tfr.keras.losses.PrecisionLambdaWeight
tfr.keras.losses.RankingLossKey
tfr.keras.losses.SigmoidCrossEntropyLoss
tfr.keras.losses.SoftmaxLoss
tfr.keras.losses.UniqueSoftmaxLoss
tfr.keras.losses.YetiDCGLambdaWeight
tfr.keras.losses.YetiLogisticLoss
tfr.keras.losses.get
tfr.keras.metrics
tfr.keras.metrics.ARPMetric
tfr.keras.metrics.AlphaDCGMetric
tfr.keras.metrics.DCGMetric
tfr.keras.metrics.HitsMetric
tfr.keras.metrics.MRRMetric
tfr.keras.metrics.MeanAveragePrecisionMetric
tfr.keras.metrics.NDCGMetric
tfr.keras.metrics.OPAMetric
tfr.keras.metrics.PrecisionIAMetric
tfr.keras.metrics.PrecisionMetric
tfr.keras.metrics.RankingMetricKey
tfr.keras.metrics.RecallMetric
tfr.keras.metrics.default_keras_metrics
tfr.keras.metrics.get
tfr.keras.model
tfr.keras.model.AbstractModelBuilder
tfr.keras.model.DNNScorer
tfr.keras.model.FeatureSpecInputCreator
tfr.keras.model.GAMScorer
tfr.keras.model.InputCreator
tfr.keras.model.ModelBuilder
tfr.keras.model.ModelBuilderWithMask
tfr.keras.model.Preprocessor
tfr.keras.model.PreprocessorWithSpec
tfr.keras.model.Scorer
tfr.keras.model.TensorDict
tfr.keras.model.TensorLike
tfr.keras.model.TypeSpecInputCreator
tfr.keras.model.UnivariateScorer
tfr.keras.model.create_keras_model
tfr.keras.pipeline
tfr.keras.pipeline.AbstractDatasetBuilder
tfr.keras.pipeline.AbstractPipeline
tfr.keras.pipeline.BaseDatasetBuilder
tfr.keras.pipeline.DatasetHparams
tfr.keras.pipeline.DatasetHparams.dataset_reader
tfr.keras.pipeline.ModelFitPipeline
tfr.keras.pipeline.MultiLabelDatasetBuilder
tfr.keras.pipeline.MultiTaskPipeline
tfr.keras.pipeline.NullDatasetBuilder
tfr.keras.pipeline.PipelineHparams
tfr.keras.pipeline.SimpleDatasetBuilder
tfr.keras.pipeline.SimplePipeline
tfr.keras.saved_model
tfr.keras.saved_model.Signatures
tfr.keras.strategy_utils
tfr.keras.strategy_utils.NullContextManager
tfr.keras.strategy_utils.get_output_filepath
tfr.keras.strategy_utils.get_strategy
tfr.keras.strategy_utils.strategy_scope
tfr.keras.utils
tfr.keras.utils.GainFunction
tfr.keras.utils.PositiveFunction
tfr.keras.utils.RankDiscountFunction
tfr.keras.utils.TensorLike
tfr.keras.utils.deserialize_keras_object
tfr.keras.utils.identity
tfr.keras.utils.inverse
tfr.keras.utils.is_greater_equal_1
tfr.keras.utils.log2_inverse
tfr.keras.utils.pow_minus_1
tfr.keras.utils.serialize_keras_object
tfr.keras.utils.symmetric_log1p
tfr.utils
tfr.utils.LossFunction
tfr.utils.MetricFunction
tfr.utils.TensorLike
tfr.utils.TransformationFunction
tfr.utils.de_noise
tfr.utils.gather_per_row
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
tfr.utils.shuffle_valid_indices
tfr.utils.sort_by_scores
tfr.utils.sorted_ranks