Index

_

__init__() (transformer_rankers.datasets.dataset.AbstractDataloader method) (transformer_rankers.datasets.dataset.QueryDocumentDataLoader method) (transformer_rankers.datasets.dataset.QueryDocumentDataset method) (transformer_rankers.datasets.dataset.QueryPosDocNegDocDataLoader method) (transformer_rankers.datasets.dataset.QueryPosDocNegDocDataset method) (transformer_rankers.datasets.dataset.WeaklySupervisedQueryDocumentDataLoader method) (transformer_rankers.datasets.dataset.WeaklySupervisedQueryDocumentDataset method) (transformer_rankers.datasets.downloader.DataDownloader method) (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) (transformer_rankers.negative_samplers.negative_sampling.BM25NegativeSamplerPyserini method) (transformer_rankers.negative_samplers.negative_sampling.RandomNegativeSampler method) (transformer_rankers.negative_samplers.negative_sampling.SentenceBERTNegativeSampler method) (transformer_rankers.scripts.train_sentenceBERT_crr.CRRBenchmarkDataReader method) (transformer_rankers.scripts.train_sentenceBERT_crr.CRRDataReader method) (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method)

A

AbstractDataloader (class in transformer_rankers.datasets.dataset) acumulate_l1_by_l2() (in module transformer_rankers.utils.utils) acumulate_list() (in module transformer_rankers.utils.utils) acumulate_list_multiple_relevant() (in module transformer_rankers.utils.utils) add_memory_hooks() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) add_module() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

adjust_logits_during_generation() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) apply() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

B

base_model() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning property) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning property) BertForPairwiseLearning (class in transformer_rankers.models.pairwise_bert) BertForPairwiseLearning.to() (in module transformer_rankers.models.pairwise_bert), [1], [2], [3] BertForPointwiseLearning (class in transformer_rankers.models.pointwise_bert) BertForPointwiseLearning.to() (in module transformer_rankers.models.pointwise_bert), [1], [2], [3] bfloat16() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

BM25NegativeSamplerPyserini (class in transformer_rankers.negative_samplers.negative_sampling) buffers() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

C

children() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) clariq_processor() (in module transformer_rankers.datasets.processors) collate_T2T_batch() (in module transformer_rankers.datasets.dataset) config_class (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning attribute) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning attribute) cpu() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

CRRBenchmarkDataReader (class in transformer_rankers.scripts.train_sentenceBERT_crr) CRRDataReader (class in transformer_rankers.scripts.train_sentenceBERT_crr) cuda() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

D

DataDownloader (class in transformer_rankers.datasets.downloader) device() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning property) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning property) double() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

download_file_from_google_drive() (in module transformer_rankers.datasets.downloader) dtype() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning property) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning property) dummy_inputs() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning property) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning property)

E

enforce_repetition_penalty_() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) eval() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

evaluate() (in module transformer_rankers.eval.results_analyses_tools) evaluate_and_aggregate() (in module transformer_rankers.eval.results_analyses_tools) evaluate_models() (in module transformer_rankers.eval.evaluation) extra_repr() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

F

fit() (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method) float() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) forward() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

from_df_to_list_without_nans() (in module transformer_rankers.utils.utils) from_pretrained() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning class method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning class method)

G

generate() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) get_confirm_token() (in module transformer_rankers.datasets.downloader) get_examples() (transformer_rankers.scripts.train_sentenceBERT_crr.CRRBenchmarkDataReader method) (transformer_rankers.scripts.train_sentenceBERT_crr.CRRDataReader method) get_extended_attention_mask() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

get_head_mask() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) get_input_embeddings() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) get_output_embeddings() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

H

half() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

I

init_weights() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

invert_attention_mask() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

L

LabelSmoothingCrossEntropy (class in transformer_rankers.models.losses.label_smoothing) LabelSmoothingCrossEntropy.to() (in module transformer_rankers.models.losses.label_smoothing), [1], [2], [3] linkso_processor() (in module transformer_rankers.datasets.processors) load_state_dict() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

load_tf_weights() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

M

main() (in module transformer_rankers.examples.aggregate_sacred_result_logs) (in module transformer_rankers.examples.analyze_ws) (in module transformer_rankers.examples.bert_ranker_cross_task_cross_ns_for_crr) (in module transformer_rankers.examples.download_task_data) (in module transformer_rankers.examples.minimal_example) (in module transformer_rankers.examples.negative_sampling) (in module transformer_rankers.examples.pairwise_bert_ranker) (in module transformer_rankers.examples.pointwise_bert_ranker) (in module transformer_rankers.examples.t5_ranker) (in module transformer_rankers.scripts.select_ns_examples) (in module transformer_rankers.scripts.train_sentenceBERT_crr) mantis_processor() (in module transformer_rankers.datasets.processors) module transformer_rankers transformer_rankers.datasets transformer_rankers.datasets.dataset transformer_rankers.datasets.downloader transformer_rankers.datasets.preprocess_crr transformer_rankers.datasets.preprocess_pr transformer_rankers.datasets.preprocess_sqr transformer_rankers.datasets.processors transformer_rankers.eval transformer_rankers.eval.evaluation transformer_rankers.eval.results_analyses_tools transformer_rankers.examples transformer_rankers.examples.aggregate_sacred_result_logs transformer_rankers.examples.analyze_ws transformer_rankers.examples.bert_ranker_cross_task_cross_ns_for_crr transformer_rankers.examples.download_task_data transformer_rankers.examples.minimal_example transformer_rankers.examples.negative_sampling transformer_rankers.examples.pairwise_bert_ranker transformer_rankers.examples.pointwise_bert_ranker transformer_rankers.examples.t5_ranker transformer_rankers.models transformer_rankers.models.losses transformer_rankers.models.losses.label_smoothing transformer_rankers.models.pairwise_bert transformer_rankers.models.pointwise_bert transformer_rankers.negative_samplers transformer_rankers.negative_samplers.negative_sampling transformer_rankers.scripts transformer_rankers.scripts.select_ns_examples transformer_rankers.scripts.train_sentenceBERT_crr transformer_rankers.trainers transformer_rankers.trainers.transformer_trainer transformer_rankers.utils transformer_rankers.utils.utils

modules() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) msdialog_processor() (in module transformer_rankers.datasets.processors)

N

named_buffers() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) named_children() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) named_modules() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

named_parameters() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) num_parameters() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

P

parameters() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) predict() (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method)

predict_with_uncertainty() (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method) prepare_inputs_for_generation() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) prune_heads() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

Q

qqp_processor() (in module transformer_rankers.datasets.processors) QueryDocumentDataLoader (class in transformer_rankers.datasets.dataset)

QueryDocumentDataset (class in transformer_rankers.datasets.dataset) QueryPosDocNegDocDataLoader (class in transformer_rankers.datasets.dataset) QueryPosDocNegDocDataset (class in transformer_rankers.datasets.dataset)

R

RandomNegativeSampler (class in transformer_rankers.negative_samplers.negative_sampling) read_crr_tsv_as_df() (in module transformer_rankers.datasets.preprocess_crr) recall_at_with_k_candidates() (in module transformer_rankers.eval.evaluation) register_backward_hook() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) register_buffer() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) register_forward_hook() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) register_forward_pre_hook() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

register_parameter() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) requires_grad_() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) reset_memory_hooks_state() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) resize_token_embeddings() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) run_experiment() (in module transformer_rankers.examples.analyze_ws) (in module transformer_rankers.examples.bert_ranker_cross_task_cross_ns_for_crr) (in module transformer_rankers.examples.pairwise_bert_ranker) (in module transformer_rankers.examples.pointwise_bert_ranker) (in module transformer_rankers.examples.t5_ranker)

S

sample() (transformer_rankers.negative_samplers.negative_sampling.BM25NegativeSamplerPyserini method) (transformer_rankers.negative_samplers.negative_sampling.RandomNegativeSampler method) (transformer_rankers.negative_samplers.negative_sampling.SentenceBERTNegativeSampler method) save_pretrained() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) save_response_content() (in module transformer_rankers.datasets.downloader)

SentenceBERTNegativeSampler (class in transformer_rankers.negative_samplers.negative_sampling) set_input_embeddings() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) state_dict() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

T

test() (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method) test_with_dropout() (transformer_rankers.trainers.transformer_trainer.TransformerTrainer method) tie_weights() (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) to() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) train() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method) transform_dstc8_to_tsv() (in module transformer_rankers.datasets.preprocess_crr) transform_linkso_to_duplicates_dfs() (in module transformer_rankers.datasets.preprocess_sqr) transform_quora_question_pairs_to_duplicates_dfs() (in module transformer_rankers.datasets.preprocess_sqr) transform_trec2020pr_to_dfs() (in module transformer_rankers.datasets.preprocess_pr) transformer_rankers module transformer_rankers.datasets module transformer_rankers.datasets.dataset module transformer_rankers.datasets.downloader module transformer_rankers.datasets.preprocess_crr module transformer_rankers.datasets.preprocess_pr module transformer_rankers.datasets.preprocess_sqr module transformer_rankers.datasets.processors module transformer_rankers.eval module transformer_rankers.eval.evaluation module transformer_rankers.eval.results_analyses_tools module transformer_rankers.examples module transformer_rankers.examples.aggregate_sacred_result_logs module transformer_rankers.examples.analyze_ws module transformer_rankers.examples.bert_ranker_cross_task_cross_ns_for_crr module

transformer_rankers.examples.download_task_data module transformer_rankers.examples.minimal_example module transformer_rankers.examples.negative_sampling module transformer_rankers.examples.pairwise_bert_ranker module transformer_rankers.examples.pointwise_bert_ranker module transformer_rankers.examples.t5_ranker module transformer_rankers.models module transformer_rankers.models.losses module transformer_rankers.models.losses.label_smoothing module transformer_rankers.models.pairwise_bert module transformer_rankers.models.pointwise_bert module transformer_rankers.negative_samplers module transformer_rankers.negative_samplers.negative_sampling module transformer_rankers.scripts module transformer_rankers.scripts.select_ns_examples module transformer_rankers.scripts.train_sentenceBERT_crr module transformer_rankers.trainers module transformer_rankers.trainers.transformer_trainer module transformer_rankers.utils module transformer_rankers.utils.utils module TransformerTrainer (class in transformer_rankers.trainers.transformer_trainer) trec2020pr_processor() (in module transformer_rankers.datasets.processors) type() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)

U

ubuntu_dstc8_processor() (in module transformer_rankers.datasets.processors)

W

WeaklySupervisedLabelSmoothingCrossEntropy (class in transformer_rankers.models.losses.label_smoothing) WeaklySupervisedLabelSmoothingCrossEntropy.to() (in module transformer_rankers.models.losses.label_smoothing), [1], [2], [3]

WeaklySupervisedQueryDocumentDataLoader (class in transformer_rankers.datasets.dataset) WeaklySupervisedQueryDocumentDataset (class in transformer_rankers.datasets.dataset)

Z

zero_grad() (transformer_rankers.models.losses.label_smoothing.LabelSmoothingCrossEntropy method) (transformer_rankers.models.losses.label_smoothing.WeaklySupervisedLabelSmoothingCrossEntropy method) (transformer_rankers.models.pairwise_bert.BertForPairwiseLearning method) (transformer_rankers.models.pointwise_bert.BertForPointwiseLearning method)