Index

A

a (test.examples.attn.test_model.TestAttentionModel attribute) (test.random.test_sample.TestSampleAfterBatching attribute) aggregate() (deletor.metrics.AggregateMetric method) (deletor.metrics.Mean method) (deletor.metrics.Sum method) AggregateMetric (class in deletor.metrics) apply_map() (in module examples.pipeline)

apply_pairwise_op() (in module deletor.losses) apply_weak_models() (deletor.models.boost.EnsembleScoringNetwork method) approx_abs() (in module deletor.math.utils) approximate_ranks() (in module deletor.ranking.utils) ApproximateBiDiNormalizedDiscountedCumulativeGain (class in deletor.losses) ApproximateNormalizedDiscountedCumulativeGain (class in deletor.losses) AttentionLayer (class in deletor.models.attn) AttentionSchedule (class in examples.attn.mltr30k)

B

bytes_feature() (in module examples.build_tfrecords)

C

call() (deletor.models.attn.AttentionLayer method) (deletor.models.attn.Encoder method) (deletor.models.attn.EncoderLayer method) (deletor.models.attn.GroupwiseAttentionNetwork method) (deletor.models.attn.GroupwiseMultiHeadAttentionNetwork method) (deletor.models.attn.MultiHeadAttention method) (deletor.models.attn.SelfAttention method) (deletor.models.gsf.GroupwiseInputNetwork method) (deletor.models.gsf.GroupwiseScoringNetwork method) (deletor.models.gsf.GroupwiseScoringNetwork2 method) (deletor.models.mlp.SimpleScoringNetwork method) check_tensor_shapes() (in module deletor.losses) compute() (deletor.losses.ApproximateBiDiNormalizedDiscountedCumulativeGain method) (deletor.losses.ApproximateNormalizedDiscountedCumulativeGain method) (deletor.losses.MeanSquaredError method) (deletor.losses.MultiLoss method) (deletor.losses.NetworkLoss method) (deletor.losses.RankingCrossEntropy method) (deletor.losses.RankingSoftmax method) (deletor.metrics.DiscountedCumulativeGain method) (deletor.metrics.Metric method) (deletor.metrics.NormalizedDiscountedCumulativeGain method)

compute_ranks() (in module deletor.ranking.utils) compute_unreduced() (deletor.losses.ApproximateBiDiNormalizedDiscountedCumulativeGain method) (deletor.losses.ApproximateNormalizedDiscountedCumulativeGain method) (deletor.losses.RankingSoftmax method) count_sampled_documents() (deletor.random.sample.DocumentSampler method) (deletor.random.sample.IndependentMultiOutputSampler method) (deletor.random.sample.IndependentSingleOutputSampler method)

D

DampedSineSchedule (class in examples.attn.mltr30k) data_file (test.examples.gsf.mltr30k.test_dataset.TestDataset attribute) dataset (test.examples.gsf.mltr30k.test_dataset.TestDataset attribute) DataTransformer (class in deletor.preprocessing) dcg() (deletor.losses.ApproximateBiDiNormalizedDiscountedCumulativeGain method) DCGLambdaWeight (class in deletor.losses) default_discount() (deletor.metrics.DiscountedCumulativeGain static method) default_dropout_rate (deletor.models.mlp.SimpleScoringNetwork attribute) default_gain() (deletor.metrics.DiscountedCumulativeGain static method) default_group_size (deletor.models.gsf.GroupwiseInputNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork2 attribute) default_share_weights (deletor.models.attn.GroupwiseMultiHeadAttentionNetwork attribute) (deletor.models.gsf.GroupwiseInputNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork2 attribute) (deletor.models.mlp.SimpleScoringNetwork attribute) default_use_average (deletor.models.attn.GroupwiseMultiHeadAttentionNetwork attribute) (deletor.models.gsf.GroupwiseInputNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork attribute) (deletor.models.gsf.GroupwiseScoringNetwork2 attribute) deletor module deletor.constants module deletor.losses module deletor.math module deletor.math.utils module deletor.metrics module

deletor.models module deletor.models.attn module deletor.models.boost module deletor.models.gsf module deletor.models.mlp module deletor.models.utils module deletor.preprocessing module deletor.random module deletor.random.sample module deletor.ranking module deletor.ranking.utils module deletor.tfutils module disable_gpu() (in module deletor.tfutils) discount_for_absolute_rank() (deletor.losses.DCGLambdaWeight method) discount_for_relative_rank_diff() (deletor.losses.DCGLambdaWeight method) DiscountedCumulativeGain (class in deletor.metrics) DocumentSampler (class in deletor.random.sample) DROPOUT_RATE (deletor.models.attn.ModelParameter attribute) (deletor.models.gsf.ModelParameter attribute) (deletor.models.mlp.ModelParameter attribute)

E

Encoder (class in deletor.models.attn) EncoderLayer (class in deletor.models.attn) EnsembleScoringNetwork (class in deletor.models.boost) evaluate() (in module examples.utils) examples module examples.attn module examples.attn.mltr30k module examples.build_tfrecords module examples.gsf module examples.gsf.mltr30k module

examples.gsf.mltr30k_boost module examples.mlp module examples.mlp.mltr30k module examples.pipeline module examples.utils module exp_rank (test.test_metrics.TestMetrics attribute) exp_softmax (test.test_losses.TestLosses attribute) exp_x_sample (test.random.test_sample.TestSampleAfterBatching attribute) exp_y_sample (test.random.test_sample.TestSampleAfterBatching attribute) expand_dims_for_unbatch() (in module examples.pipeline) expected_dcg() (test.test_metrics.TestMetrics class method)

F

float_feature() (in module examples.build_tfrecords)

G

get_config() (examples.attn.mltr30k.AttentionSchedule method) (examples.attn.mltr30k.DampedSineSchedule method) get_number_of_samples() (deletor.random.sample.IndependentMultiOutputSampler method) (deletor.random.sample.IndependentSingleOutputSampler method) get_valid_pairs_and_clean_labels() (in module deletor.losses) GROUP_SIZE (deletor.models.attn.ModelParameter attribute) (deletor.models.gsf.ModelParameter attribute)

group_size (test.examples.gsf.mltr30k.test_model.MyTestCase attribute) GroupwiseAttentionNetwork (class in deletor.models.attn) GroupwiseInputNetwork (class in deletor.models.gsf) GroupwiseMultiHeadAttentionNetwork (class in deletor.models.attn) GroupwiseScoringNetwork (class in deletor.models.gsf) GroupwiseScoringNetwork2 (class in deletor.models.gsf) grow_memory() (in module deletor.tfutils)

I

IndependentMultiOutputSampler (class in deletor.random.sample) IndependentSingleOutputSampler (class in deletor.random.sample) individual_weights() (deletor.losses.DCGLambdaWeight method) (deletor.losses.LambdaWeight method) int64_feature() (in module examples.build_tfrecords) inverse_max_dcg() (in module deletor.ranking.utils)

inverse_transform() (deletor.preprocessing.DataTransformer method) (deletor.preprocessing.MinMaxScaler method) (deletor.preprocessing.PowerTransformer method) (deletor.preprocessing.RobustScaler method) (deletor.preprocessing.StandardScaler method) is_label_valid() (in module deletor.ranking.utils) is_valid_query() (in module examples.pipeline)

L

LambdaWeight (class in deletor.losses) LIST_SIZE (deletor.models.mlp.ModelParameter attribute)

ListwiseLoss (class in deletor.losses) load_dataset() (in module examples.pipeline) log_level_type() (in module examples.utils)

M

main() (in module examples.attn.mltr30k) (in module examples.gsf.mltr30k) (in module examples.gsf.mltr30k_boost) (in module examples.mlp.mltr30k) make_attention_mask() (deletor.models.attn.MultiHeadAttention class method) make_callbacks() (in module examples.gsf.mltr30k_boost) make_command_line_options() (in module examples.attn.mltr30k) (in module examples.build_tfrecords) (in module examples.gsf.mltr30k) (in module examples.mlp.mltr30k) make_dataset() (in module examples.build_tfrecords) make_example() (in module examples.build_tfrecords) make_feature_description() (in module examples.pipeline) make_gather_indices() (deletor.random.sample.DocumentSampler method) (deletor.random.sample.IndependentMultiOutputSampler method) make_loss() (in module examples.utils) make_optimizer() (in module examples.utils) make_padded_shapes() (in module examples.pipeline) make_padding_values() (in module examples.pipeline) make_point_wise_feed_forward_network() (in module deletor.models.attn) make_rank_indices() (deletor.metrics.DiscountedCumulativeGain class method) make_scaler() (in module examples.pipeline) make_scatter_indices() (deletor.random.sample.DocumentSampler method) (deletor.random.sample.IndependentMultiOutputSampler method) (deletor.random.sample.IndependentSingleOutputSampler method) make_shared_layer() (deletor.models.attn.GroupwiseAttentionNetwork class method) masked_softmax() (in module deletor.losses) Mean (class in deletor.metrics) MeanSquaredError (class in deletor.losses) Metric (class in deletor.metrics) MinMaxScaler (class in deletor.preprocessing) ModelParameter (class in deletor.models.attn) (class in deletor.models.boost) (class in deletor.models.gsf) (class in deletor.models.mlp) module deletor deletor.constants deletor.losses deletor.math deletor.math.utils deletor.metrics deletor.models deletor.models.attn deletor.models.boost deletor.models.gsf deletor.models.mlp deletor.models.utils deletor.preprocessing deletor.random deletor.random.sample deletor.ranking deletor.ranking.utils deletor.tfutils examples examples.attn examples.attn.mltr30k examples.build_tfrecords examples.gsf examples.gsf.mltr30k examples.gsf.mltr30k_boost examples.mlp examples.mlp.mltr30k examples.pipeline examples.utils test test.examples test.examples.attn test.examples.attn.test_model test.examples.gsf test.examples.gsf.mltr30k test.examples.gsf.mltr30k.test_dataset test.examples.gsf.mltr30k.test_model test.examples.mlp test.examples.mlp.mltr30k test.examples.mlp.mltr30k.test_models test.examples.test_pipeline test.random test.random.test_sample test.ranking test.ranking.test_utils test.test_losses test.test_metrics test.test_preprocessing

MultiHeadAttention (class in deletor.models.attn) MultiLoss (class in deletor.losses) multiple (test.examples.gsf.mltr30k.test_model.MyTestCase attribute) MyTestCase (class in test.examples.gsf.mltr30k.test_model)

N

N_FEATURES (deletor.models.attn.ModelParameter attribute) (deletor.models.gsf.ModelParameter attribute) n_features (test.examples.attn.test_model.TestAttentionModel attribute) N_FF_UNITS (deletor.models.attn.ModelParameter attribute) N_HEADS (deletor.models.attn.ModelParameter attribute) N_LAYERS (deletor.models.attn.ModelParameter attribute) N_MODEL (deletor.models.attn.ModelParameter attribute) N_SAMPLED_FEATURES (deletor.models.boost.ModelParameter attribute) N_UNITS (deletor.models.gsf.ModelParameter attribute) (deletor.models.mlp.ModelParameter attribute)

name() (deletor.losses.ApproximateNormalizedDiscountedCumulativeGain method) (deletor.losses.ListwiseLoss property) (deletor.metrics.AggregateMetric property) (deletor.metrics.DiscountedCumulativeGain property) (deletor.metrics.Metric property) (deletor.metrics.NormalizedDiscountedCumulativeGain property) NetworkLoss (class in deletor.losses) normalize_dropout() (in module deletor.models.utils) normalize_weights() (deletor.losses.ListwiseLoss method) (deletor.losses.NetworkLoss method) NormalizedDiscountedCumulativeGain (class in deletor.metrics)

P

pad_groups() (in module examples.pipeline) padded_shapes (test.examples.gsf.mltr30k.test_model.MyTestCase attribute) padding_values (test.examples.gsf.mltr30k.test_model.MyTestCase attribute) pair_weights() (deletor.losses.DCGLambdaWeight method) (deletor.losses.LambdaWeight method) pairwise() (in module test.examples.gsf.mltr30k.test_dataset) parse_example() (in module examples.pipeline) PowerTransformer (class in deletor.preprocessing)

precompute() (deletor.losses.RankingSoftmax method) prepare_data() (in module examples.attn.mltr30k) (in module examples.gsf.mltr30k) (in module examples.gsf.mltr30k_boost) (in module examples.mlp.mltr30k) prepare_strong_data() (in module examples.gsf.mltr30k_boost) prepare_test_data() (in module examples.gsf.mltr30k_boost) prepare_weak_data() (in module examples.gsf.mltr30k_boost)

R

r (test.examples.attn.test_model.TestAttentionModel attribute) (test.random.test_sample.TestSampleAfterBatching attribute) RANDOM_SEED (deletor.models.gsf.ModelParameter attribute) (deletor.models.mlp.ModelParameter attribute) rank_by_scores() (deletor.metrics.DiscountedCumulativeGain class method)

RankingCrossEntropy (class in deletor.losses) RankingSoftmax (class in deletor.losses) reshape_input() (deletor.models.attn.MultiHeadAttention class method) reshape_output() (deletor.models.attn.MultiHeadAttention class method) RobustScaler (class in deletor.preprocessing)

S

sample() (deletor.random.sample.DocumentSampler method) sample_after_batching() (deletor.random.sample.DocumentSampler method) (deletor.random.sample.IndependentMultiOutputSampler method) sample_before_batching() (deletor.random.sample.DocumentSampler method) (deletor.random.sample.IndependentMultiOutputSampler method) scaled_dot_product_attention() (deletor.models.attn.MultiHeadAttention class method) seed (test.examples.gsf.mltr30k.test_model.MyTestCase attribute) select_features() (in module examples.pipeline) SelfAttention (class in deletor.models.attn) set_device() (in module deletor.tfutils) setup_model() (in module examples.attn.mltr30k) (in module examples.gsf.mltr30k) (in module examples.gsf.mltr30k_boost) (in module examples.mlp.mltr30k) setUpClass() (test.examples.attn.test_model.TestAttentionModel class method) (test.examples.gsf.mltr30k.test_dataset.TestDataset class method) (test.examples.gsf.mltr30k.test_model.MyTestCase class method) (test.examples.test_pipeline.TestPipeline class method)

SHARE_WEIGHTS (deletor.models.attn.ModelParameter attribute) (deletor.models.gsf.ModelParameter attribute) shuffle_documents() (in module examples.pipeline) SimpleScoringNetwork (class in deletor.models.mlp) sort_by_scores() (in module deletor.ranking.utils) sorted_ranks() (in module deletor.ranking.utils) spacing() (in module deletor.math.utils) split_heads() (deletor.models.attn.MultiHeadAttention method) squeeze_for_unbatch() (in module examples.pipeline) StandardScaler (class in deletor.preprocessing) stats_file (test.examples.gsf.mltr30k.test_dataset.TestDataset attribute) Sum (class in deletor.metrics) svm_generator() (in module examples.build_tfrecords)

T

test module test.examples module test.examples.attn module test.examples.attn.test_model module test.examples.gsf module test.examples.gsf.mltr30k module test.examples.gsf.mltr30k.test_dataset module test.examples.gsf.mltr30k.test_model module test.examples.mlp module test.examples.mlp.mltr30k module test.examples.mlp.mltr30k.test_models module test.examples.test_pipeline module test.random module test.random.test_sample module test.ranking module test.ranking.test_utils module test.test_losses module test.test_metrics module test.test_preprocessing module test_approximate_bidi_ndcg() (test.test_losses.TestLosses method) test_approximate_ndcg() (test.test_losses.TestLosses method) test_call() (test.examples.attn.test_model.TestAttentionModel method) test_compute_ranks() (test.ranking.test_utils.TestUtils method) test_dataset() (test.examples.gsf.mltr30k.test_dataset.TestDataset method) test_dcg() (test.test_metrics.TestMetrics method) test_gin_forward_pass() (test.examples.gsf.mltr30k.test_model.MyTestCase method) test_gsf_foward_pass() (test.examples.gsf.mltr30k.test_model.MyTestCase method) test_independent_multi_output() (test.random.test_sample.TestSampleAfterBatching method) (test.random.test_sample.TestSampleBeforeBatching method) test_independent_single_output() (test.random.test_sample.TestSampleAfterBatching method) test_masked_softmax() (test.test_losses.TestLosses method)

test_mean_squared_error() (test.test_losses.TestLosses method) test_min_max_scaler() (test.test_preprocessing.TestPreprocessing method) test_multi_loss() (test.test_losses.TestLosses method) test_ndcg() (test.test_metrics.TestMetrics method) test_power_transformer() (test.test_preprocessing.TestPreprocessing method) test_ranking_cross_entropy() (test.test_losses.TestLosses method) test_ranking_softmax_loss() (test.test_losses.TestLosses method) test_robust_scaler() (test.test_preprocessing.TestPreprocessing method) test_sample_documents_flat_indices() (test.examples.gsf.mltr30k.test_dataset.TestDataset method) test_sample_windowed() (test.examples.gsf.mltr30k.test_dataset.TestDataset method) test_scaled_dot_product_attention() (test.examples.attn.test_model.TestAttentionModel method) test_scaled_dot_product_attention_with_weights() (test.examples.attn.test_model.TestAttentionModel method) test_shuffle_documents() (test.examples.gsf.mltr30k.test_dataset.TestDataset method) test_simple_scoring_network() (test.examples.mlp.mltr30k.test_models.TestModels method) test_standard_scaler() (test.test_preprocessing.TestPreprocessing method) test_tfr_approximate_ndcg() (test.test_losses.TestLosses method) test_tfr_create_ndcg_lambda_weight() (test.test_losses.TestLambdaWeights method) test_tfr_dcg() (test.test_metrics.TestMetrics method) test_tfr_default() (test.test_losses.TestLambdaWeights method) test_tfr_gain_and_discount() (test.test_losses.TestLambdaWeights method) test_tfr_individual_weights() (test.test_losses.TestLambdaWeights method) test_tfr_invalid_labels() (test.test_losses.TestLambdaWeights method) test_tfr_ndcg() (test.test_metrics.TestMetrics method) test_tfr_ndcg_with_weights() (test.test_metrics.TestMetrics method) test_tfr_ndcg_with_weights_zero_relevance() (test.test_metrics.TestMetrics method) test_tfr_ndcg_with_zero_relevance() (test.test_metrics.TestMetrics method) test_tfr_normalized() (test.test_losses.TestLambdaWeights method) test_tfr_smooth_fraction() (test.test_losses.TestLambdaWeights method) test_tfr_topn() (test.test_losses.TestLambdaWeights method) test_truncate_list_size() (test.examples.gsf.mltr30k.test_dataset.TestDataset method) test_unbatch() (test.examples.test_pipeline.TestPipeline method) TestAttentionModel (class in test.examples.attn.test_model) TestDataset (class in test.examples.gsf.mltr30k.test_dataset) TestLambdaWeights (class in test.test_losses) TestLosses (class in test.test_losses) TestMetrics (class in test.test_metrics) TestModels (class in test.examples.mlp.mltr30k.test_models) TestPipeline (class in test.examples.test_pipeline) TestPreprocessing (class in test.test_preprocessing) TestSampleAfterBatching (class in test.random.test_sample) TestSampleBeforeBatching (class in test.random.test_sample) TestUtils (class in test.ranking.test_utils) to_nd_indices() (in module deletor.tfutils) train() (in module examples.utils) transform() (deletor.preprocessing.DataTransformer method) (deletor.preprocessing.MinMaxScaler method) (deletor.preprocessing.PowerTransformer method) (deletor.preprocessing.RobustScaler method) (deletor.preprocessing.StandardScaler method) truncate_document_list() (in module examples.pipeline)

U

update_weights() (in module examples.gsf.mltr30k_boost) USE_AVERAGE (deletor.models.attn.ModelParameter attribute) (deletor.models.gsf.ModelParameter attribute)

USE_LAYER_NORM (deletor.models.attn.ModelParameter attribute)

V

validate_inputs() (deletor.losses.NetworkLoss static method) (deletor.metrics.Metric static method)

W

WeakGroupwiseScoringNetwork (class in deletor.models.boost)

write_data() (in module examples.build_tfrecords) write_scalers() (in module examples.build_tfrecords)

X

x (test.examples.attn.test_model.TestAttentionModel attribute) (test.random.test_sample.TestSampleAfterBatching attribute) (test.random.test_sample.TestSampleBeforeBatching attribute) (test.test_preprocessing.TestPreprocessing attribute)

Y

y (test.examples.attn.test_model.TestAttentionModel attribute) (test.random.test_sample.TestSampleAfterBatching attribute) (test.random.test_sample.TestSampleBeforeBatching attribute) y_pred (test.test_losses.TestLosses attribute) (test.test_metrics.TestMetrics attribute)

y_pred_dense (test.test_metrics.TestMetrics attribute) y_true (test.test_losses.TestLosses attribute) (test.test_metrics.TestMetrics attribute)