Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | Z _ __add__() (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing method) (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel method) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel method) __call__() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.data_models.data_collator.FiDDataCollator method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) (primeqa.qg.processors.passage_qg.qg_processor.QGProcessor method) (primeqa.qg.processors.table_qg.sql_processor.SqlProcessor method) (primeqa.qg.utils.data_collator.T2TDataCollator method) (primeqa.tableqa.utils.data_collator.TapasCollator method) __mul__() (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing method) (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel method) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel method) A AbstractPostProcessor (class in primeqa.mrc.processors.postprocessors.abstract) AbstractPreProcessor (class in primeqa.mrc.processors.preprocessors.abstract) AbstractTaskHead (class in primeqa.mrc.models.heads.abstract) adapt_dataset() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) add() (primeqa.mrc.metrics.mlqa.mlqa.MLQA method) (primeqa.mrc.metrics.nq_f1.nq_f1.NQF1 method) (primeqa.mrc.metrics.squad.squad.SQUAD method) (primeqa.mrc.metrics.tydi_f1.tydi_f1.TyDiF1 method) add_batch() (primeqa.mrc.metrics.mlqa.mlqa.MLQA method) (primeqa.mrc.metrics.nq_f1.nq_f1.NQF1 method) (primeqa.mrc.metrics.squad.squad.SQUAD method) (primeqa.mrc.metrics.tydi_f1.tydi_f1.TyDiF1 method) add_callback() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) add_column_types() (primeqa.qg.models.table_qg.sql_sampler.SimpleSqlSampler static method) add_document_text_to_hit() (primeqa.ir.util.corpus_reader.DocumentCollection method) add_memory_hooks() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) add_module() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) adjust_logits_during_generation() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) align() (in module primeqa.ir.dense.colbert_top.colbert.training.training) all_gather() (in module primeqa.util.transformers_utils.torch_utils) ANNIndex (class in primeqa.ir.dense.dpr_top.dpr.faiss_index) AnswerSampler (class in primeqa.qg.models.passage_qg.answer_sampler) apply() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) Arguments (class in primeqa.ir.dense.colbert_top.colbert.utils.parser) assign_label_to_passage() (in module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers) autocast_smart_context_manager() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) B barrier() (in module primeqa.ir.dense.colbert_top.colbert.utils.distributed) base_model (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) BaseColBERT (class in primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert) BasePreProcessor (class in primeqa.mrc.processors.preprocessors.base) batch() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) batch_dict() (primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder.BiEncoderLoader method) batch_retrieve() (primeqa.ir.sparse.retriever.PyseriniRetriever method) beam_sample() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) beam_search() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) bfloat16() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) BiEncoder (class in primeqa.ir.dense.dpr_top.dpr.biencoder_gcp) BiEncoderBatches (class in primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder) BiEncoderHypers (class in primeqa.ir.dense.dpr_top.dpr.biencoder_hypers) BiEncoderInst (class in primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder) BiEncoderLoader (class in primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder) BiEncoderTrainArgs (class in primeqa.ir.dense.dpr_top.dpr.biencoder_trainer) BiEncoderTrainer (class in primeqa.ir.dense.dpr_top.dpr.biencoder_trainer) block_shuffle() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) BM25Config (class in primeqa.ir.sparse.config) BM25Engine (class in primeqa.ir.sparse.bm25_engine) buffers() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) build_index() (in module primeqa.ir.dense.dpr_top.dpr.faiss_index) byte_slice() (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) C calculate_distance() (in module primeqa.ir.dense.colbert_top.colbert.training.training) CandidateGeneration (class in primeqa.ir.dense.colbert_top.colbert.search.candidate_generation) check_sizes() (in module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers) Checkpoint (class in primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint) checkpoint_on_exit() (in module primeqa.ir.dense.colbert_top.colbert.utils.signals) children() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) clear() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) ColBERT (class in primeqa.ir.dense.colbert_top.colbert.modeling.colbert) colbert_score() (in module primeqa.ir.dense.colbert_top.colbert.modeling.colbert) colbert_score_packed() (in module primeqa.ir.dense.colbert_top.colbert.modeling.colbert) colbert_score_reduce() (in module primeqa.ir.dense.colbert_top.colbert.modeling.colbert) ColBERTConfig (class in primeqa.ir.dense.colbert_top.colbert.infra.config) Collection (class in primeqa.ir.dense.colbert_top.colbert.data.collection) CollectionEncoder (class in primeqa.ir.dense.colbert_top.colbert.indexing.collection_encoder) CollectionIndexer (class in primeqa.ir.dense.colbert_top.colbert.indexing.collection_indexer) compress_into_codes() (primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual.ResidualCodec method) compute() (primeqa.mrc.metrics.mlqa.mlqa.MLQA method) (primeqa.mrc.metrics.nq_f1.nq_f1.NQF1 method) (primeqa.mrc.metrics.squad.squad.SQUAD method) (primeqa.mrc.metrics.tydi_f1.tydi_f1.TyDiF1 method) compute_and_write_labels() (in module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers) compute_denotation_accuracy() (in module primeqa.tableqa.metrics.answer_accuracy) compute_erm_loss() (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) compute_f1() (in module primeqa.mrc.metrics.nq_f1.nq_eval) compute_faiss_kmeans() (in module primeqa.ir.dense.colbert_top.colbert.indexing.collection_indexer) compute_final_f1() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) compute_loss() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) compute_macro_f1() (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) compute_optimal_metrics() (in module primeqa.mrc.metrics.nq_f1.nq_eval) compute_partial_match_scores() (in module primeqa.mrc.metrics.tydi_f1.eval_utils) compute_pr_curves() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) compute_score_diff_between_span_and_cls() (in module primeqa.mrc.processors.postprocessors.scorers) compute_short_answer_type_weighted_score_diff_between_span_and_cls() (in module primeqa.mrc.processors.postprocessors.scorers) compute_transition_beam_scores() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) compute_weighted_sum_short_answer_type_score_diff_between_span_and_cls() (in module primeqa.mrc.processors.postprocessors.scorers) ConfidenceScorer (class in primeqa.calibration.confidence_scorer) config_class (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT attribute) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta attribute) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR attribute) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker attribute) constrained_beam_search() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) convert_sql_to_string() (primeqa.qg.models.table_qg.sql_sampler.SimpleSqlSampler method) convert_to_features() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) convert_to_float() (in module primeqa.tableqa.tapex.utils.wikisql_utils_for_tapex) copy() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) Corpus (class in primeqa.ir.dense.dpr_top.dpr.simple_mmap_dataset) corpus_reader() (in module primeqa.ir.util.corpus_reader) count() (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing method) (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel method) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel method) cpu() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) create_conflict_free_batches() (primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder.BiEncoderBatches method) create_dataset_from_json_str() (in module primeqa.boolqa.processors.dataset.mrc2dataset) create_dataset_from_run_mrc_output() (in module primeqa.boolqa.processors.dataset.mrc2dataset) create_directory() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) create_optimizer() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) create_optimizer_and_scheduler() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) create_qg_input() (primeqa.qg.models.hybrid_qg.path_sampler.PathSampler method) (primeqa.qg.models.passage_qg.answer_sampler.AnswerSampler method) create_scheduler() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) cuda() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) D Dataset (class in primeqa.ir.dense.colbert_top.colbert.data.dataset) DatasetProcessor (class in primeqa.tableqa.preprocessors.dataset) DataTrainingArguments (class in primeqa.calibration.train_confidence_calibrator) (class in primeqa.qg.run_qg) (class in primeqa.tableqa.tapex.utils.argument_utils_for_tapex) decompress() (primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual.ResidualCodec method) detect_language() (primeqa.qg.models.passage_qg.answer_sampler.AnswerSampler method) device (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) dist_initialize() (in module primeqa.ir.dense.dpr_top.torch_util.hypers_base) (in module primeqa.util.transformers_utils.hypers_base) DistBatchesBase (class in primeqa.ir.dense.dpr_top.dataloader.distloader_base) (class in primeqa.util.dataloader.distloader_base) Distribution (class in primeqa.ir.dense.dpr_top.util.reporting) DocTokenizer (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization) DocTokenizerRoberta (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_roberta) DocTokenizerXLMR (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_xlmr) DocumentCollection (class in primeqa.ir.util.corpus_reader) dotdict (class in primeqa.ir.dense.colbert_top.colbert.utils.utils) dotdict_lax (class in primeqa.ir.dense.colbert_top.colbert.utils.utils) double() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) download_and_prepare() (primeqa.mrc.metrics.mlqa.mlqa.MLQA method) (primeqa.mrc.metrics.nq_f1.nq_f1.NQF1 method) (primeqa.mrc.metrics.squad.squad.SQUAD method) (primeqa.mrc.metrics.tydi_f1.tydi_f1.TyDiF1 method) DPRIndexer (class in primeqa.ir.dense.dpr_top.dpr.index_simple_corpus) DPRIndexingArguments (class in primeqa.ir.dense.dpr_top.dpr.config) DPROptions (class in primeqa.ir.dense.dpr_top.dpr.dpr_util) DPRSearchArguments (class in primeqa.ir.dense.dpr_top.dpr.config) DPRSearcher (class in primeqa.ir.dense.dpr_top.dpr.searcher) DPRTrainingArguments (class in primeqa.ir.dense.dpr_top.dpr.config) dtype (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) dummy_inputs (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) E EagerBatcher (class in primeqa.ir.dense.colbert_top.colbert.training.eager_batcher_v2) ElectraReranker (class in primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra) ELI5FiDPostProcessor (class in primeqa.mrc.processors.postprocessors.eli5_fid) ELI5FiDPreprocessor (class in primeqa.mrc.processors.preprocessors.eli5_fid) encode() (in module primeqa.ir.dense.colbert_top.colbert.indexing.collection_indexer) encode_passages() (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) EncoderWrapper (class in primeqa.ir.dense.dpr_top.dpr.biencoder_gcp) (class in primeqa.mrc.models.fid_task_model) estimate_tokens() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) eval() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) eval_metrics (primeqa.calibration.train_confidence_calibrator.TaskArguments attribute) EvalPredictionWithProcessing (class in primeqa.mrc.data_models.eval_prediction_with_processing) evaluate() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) (in module primeqa.mrc.metrics.squad.evaluate) (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) evaluate_recall() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.metrics) evaluation_loop() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) exact_match_score() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) (in module primeqa.mrc.metrics.squad.evaluate) example_id (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) Examples (class in primeqa.ir.dense.colbert_top.colbert.data.examples) expand_files() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) extra_repr() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) extract_metrics_at_optimal_threshold() (in module primeqa.mrc.metrics.nq_f1.nq_eval) ExtractivePostProcessor (class in primeqa.mrc.processors.postprocessors.extractive) ExtractiveQAHead (class in primeqa.mrc.models.heads.extractive) ExtractiveQAModelOutput (class in primeqa.mrc.data_models.model_outputs.extractive) ExtractiveQAWithConfidenceHead (class in primeqa.mrc.models.heads.extractive) ExtractiveQAWithConfidenceModelOutput (class in primeqa.mrc.data_models.model_outputs.extractive) F f1_score() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) (in module primeqa.mrc.metrics.squad.evaluate) f7() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) FiDDataCollator (class in primeqa.mrc.data_models.data_collator) FiDModelForDownstreamTasks (class in primeqa.mrc.models.fid_task_model) file_tqdm() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) fill_from_args() (in module primeqa.ir.dense.dpr_top.util.args_help) (in module primeqa.util.args_helper) (primeqa.ir.dense.dpr_top.dpr.biencoder_hypers.BiEncoderHypers method) (primeqa.ir.dense.dpr_top.dpr.biencoder_trainer.BiEncoderTrainArgs method) (primeqa.ir.dense.dpr_top.dpr.dpr_util.DPROptions method) (primeqa.ir.dense.dpr_top.dpr.searcher.Options method) (primeqa.ir.dense.dpr_top.torch_util.hypers_base.HypersBase method) fill_from_config() (in module primeqa.ir.dense.dpr_top.util.args_help) (primeqa.ir.dense.dpr_top.dpr.biencoder_hypers.BiEncoderHypers method) (primeqa.ir.dense.dpr_top.dpr.biencoder_trainer.BiEncoderTrainArgs method) (primeqa.ir.dense.dpr_top.dpr.dpr_util.DPROptions method) (primeqa.ir.dense.dpr_top.dpr.searcher.Options method) (primeqa.ir.dense.dpr_top.torch_util.hypers_base.HypersBase method) fill_from_dict() (in module primeqa.ir.dense.dpr_top.util.args_help) (in module primeqa.util.args_helper) flatten() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) float() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) floating_point_ops() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) format_metadata() (in module primeqa.ir.dense.colbert_top.utility.utils.save_metadata) forward() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) framework (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) from_bool_label() (primeqa.mrc.data_models.target_type.TargetType class method) from_config() (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks class method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks class method) from_pretrained() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT class method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta class method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR class method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker class method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks class method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks class method) fromkeys() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) G generate() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) get_annotations() (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) get_buffer() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_candidate_hybrid_chains() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) get_colbert_from_pretrained() (in module primeqa.ir.dense.colbert_top.colbert.modeling.factory) get_deltas() (in module primeqa.ir.dense.colbert_top.colbert.indexing.loaders) get_dim_and_nbits() (in module primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings) get_doc_tokenizer() (in module primeqa.ir.dense.colbert_top.colbert.modeling.factory) get_eval_dataloader() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) get_extended_attention_mask() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_extra_state() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_head_mask() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_input_embeddings() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_language_id() (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.sparse.utils) get_latex_str() (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) get_metadata() (in module primeqa.ir.dense.colbert_top.utility.utils.save_metadata) get_metadata_only() (in module primeqa.ir.dense.colbert_top.utility.utils.save_metadata) get_metrics_as_dict() (in module primeqa.mrc.metrics.nq_f1.nq_eval) get_metrics_with_answer_stats() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) get_model_type() (in module primeqa.ir.dense.colbert_top.colbert.modeling.factory) get_named_entities() (primeqa.qg.models.passage_qg.answer_sampler.AnswerSampler method) get_optimizer_cls_and_kwargs() (primeqa.mrc.trainers.mrc.MRCTrainer static method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer static method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer static method) (primeqa.qg.trainers.qg_trainer.QGTrainer static method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer static method) get_output_embeddings() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_parameter() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_parts() (in module primeqa.ir.dense.colbert_top.colbert.indexing.loaders) get_predicted_answer_text() (primeqa.tableqa.postprocessor.wikisql.WikiSQLPostprocessor method) get_query_tokenizer() (in module primeqa.ir.dense.colbert_top.colbert.modeling.factory) get_raw_datasets() (in module primeqa.mrc.run_mrc_utils) get_submodule() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) get_supported() (primeqa.mrc.processors.postprocessors.scorers.SupportedSpanScorers class method) get_test_dataloader() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) get_train_dataloader() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) gold_has_long_answer() (in module primeqa.mrc.metrics.nq_f1.eval_utils) gold_has_minimal_answer() (in module primeqa.mrc.metrics.tydi_f1.eval_utils) gold_has_passage_answer() (in module primeqa.mrc.metrics.tydi_f1.eval_utils) gold_has_short_answer() (in module primeqa.mrc.metrics.nq_f1.eval_utils) gradient_checkpointing_disable() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) gradient_checkpointing_enable() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) greedy_search() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) group_beam_search() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) groupby_first_item() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) grouper() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) gunzip_str() (in module primeqa.ir.dense.dpr_top.dpr.simple_mmap_dataset) (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) gzip_str() (in module primeqa.ir.dense.dpr_top.dpr.simple_mmap_dataset) (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) H half() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) handle_args() (in module primeqa.ir.run_bm25_retrieval) HF_ColBERT (class in primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert) HF_ColBERT_Roberta (class in primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta) HF_ColBERT_XLMR (class in primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr) hybrid_chain_to_t5_sequence() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) hybrid_chains() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) HybridQAProcessor (class in primeqa.qg.processors.hybrid_qg.hybridqa_processor) hyperparameter_search() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) HypersBase (class in primeqa.ir.dense.dpr_top.torch_util.hypers_base) (class in primeqa.util.transformers_utils.hypers_base) I InconsistentSpanError index() (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing method) (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel method) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel method) Indexer (class in primeqa.ir.dense.colbert_top.colbert.indexer) IndexingArguments (class in primeqa.ir.sparse.config) IndexLoader (class in primeqa.ir.dense.colbert_top.colbert.search.index_loader) IndexManager (class in primeqa.ir.dense.colbert_top.colbert.indexing.index_manager) IndexOptions (class in primeqa.ir.dense.dpr_top.dpr.faiss_index) IndexSaver (class in primeqa.ir.dense.colbert_top.colbert.indexing.index_saver) IndexScorer (class in primeqa.ir.dense.colbert_top.colbert.search.index_storage) IndividualDomainBatchSampler (class in primeqa.mrc.trainers.mrc_mskd) InferenceArguments (class in primeqa.qg.run_qg) info (primeqa.mrc.metrics.mlqa.mlqa.MLQA property) (primeqa.mrc.metrics.nq_f1.nq_f1.NQF1 property) (primeqa.mrc.metrics.squad.squad.SQUAD property) (primeqa.mrc.metrics.tydi_f1.tydi_f1.TyDiF1 property) init() (in module primeqa.ir.dense.colbert_top.colbert.utils.distributed) init_git_repo() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) init_weights() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) initialize_scorer() (in module primeqa.mrc.processors.postprocessors.scorers) int_or_float() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) invert_attention_mask() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) is_csv() (in module primeqa.ir.util.corpus_reader) is_gradient_checkpointing (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta property) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR property) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker property) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks property) (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) is_local_process_zero() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) is_null_span() (primeqa.mrc.metrics.nq_f1.eval_utils.NQSpan method) (primeqa.mrc.metrics.tydi_f1.eval_utils.Span method) is_null_span_list() (in module primeqa.mrc.metrics.nq_f1.eval_utils) is_tsv() (in module primeqa.ir.util.corpus_reader) is_world_process_zero() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) items() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) J jsonl_files() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) jsonl_instances() (primeqa.ir.dense.dpr_top.dpr.biencoder_hypers.BiEncoderHypers method) (primeqa.ir.dense.dpr_top.dpr.biencoder_trainer.BiEncoderTrainArgs method) (primeqa.ir.dense.dpr_top.dpr.dpr_util.DPROptions method) (primeqa.ir.dense.dpr_top.dpr.searcher.Options method) (primeqa.ir.dense.dpr_top.torch_util.hypers_base.HypersBase method) jsonl_lines() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) jsonl_records() (in module primeqa.ir.dense.dpr_top.util.line_corpus) K keys() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) kofn() (primeqa.util.transformers_utils.hypers_base.HypersBase method) L l2_convert_indexed_vectors() (in module primeqa.ir.dense.dpr_top.dpr.faiss_index) l2_convert_query_vectors() (in module primeqa.ir.dense.dpr_top.dpr.faiss_index) label_features_for_subsampling() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) label_ids (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing attribute) language (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) Launcher (class in primeqa.ir.dense.colbert_top.colbert.infra.launcher) LazyBatcher (class in primeqa.ir.dense.colbert_top.colbert.training.lazy_batcher) lengths2offsets() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) link_sents_to_cells() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) list_corpus_files() (in module primeqa.ir.util.corpus_reader) load_batch_backgrounds() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) load_checkpoint() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) load_checkpoint_raw() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) load_colbert() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) load_collection() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) load_collection_() (in module primeqa.ir.dense.colbert_top.utility.utils.qa_loaders) load_compressed_index_part() (in module primeqa.ir.dense.colbert_top.colbert.indexing.index_manager) load_corpus() (primeqa.ir.util.corpus_reader.DocumentCollection method) load_data() (primeqa.tableqa.preprocessors.dataset.TableQADataset method) load_doclens() (in module primeqa.ir.dense.colbert_top.colbert.indexing.loaders) load_gt_lookup_as_dict() (in module primeqa.mrc.metrics.nq_f1.nq_eval) load_index_part() (in module primeqa.ir.dense.colbert_top.colbert.indexing.index_manager) load_model() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.load_model) load_only_extended_model() (in module primeqa.util.transformers_utils.model_utils) load_pretrained() (in module primeqa.util.transformers_utils.model_utils) load_qas_() (in module primeqa.ir.dense.colbert_top.utility.utils.qa_loaders) load_qrels() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) load_queries() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.sparse.utils) load_ranking() (in module primeqa.ir.dense.colbert_top.colbert.data.ranking) (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) load_state_dict() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) load_tf_weights() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) load_tokenizer() (in module primeqa.util.transformers_utils.model_utils) load_topK() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) load_topK_pids() (in module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders) log() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) log_metrics() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) Logger (class in primeqa.ir.dense.colbert_top.colbert.utils.logging) long_answer_span (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) long_score (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) lookup_by_aliases() (in module primeqa.ir.util.corpus_reader) lookup_centroids() (primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual.ResidualCodec method) LossHistory (class in primeqa.ir.dense.dpr_top.torch_util.transformer_optimize) (class in primeqa.util.transformers_utils.optimizer_utils) M main() (in module primeqa.boolqa.run_score_normalizer) (in module primeqa.calibration.train_confidence_calibrator) (in module primeqa.ir.dense.colbert_top.colbert.run_indexer) (in module primeqa.ir.dense.colbert_top.colbert.run_searcher) (in module primeqa.ir.dense.colbert_top.colbert.run_trainer) (in module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM) (in module primeqa.ir.dense.colbert_top.utility.preprocess.docs2passages) (in module primeqa.ir.dense.colbert_top.utility.rankings.dev_subsample) (in module primeqa.ir.dense.colbert_top.utility.rankings.merge) (in module primeqa.ir.dense.colbert_top.utility.rankings.split_by_offset) (in module primeqa.ir.dense.colbert_top.utility.rankings.split_by_queries) (in module primeqa.ir.dense.colbert_top.utility.rankings.tune) (in module primeqa.ir.dense.dpr_top.dpr.biencoder_trainer) (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.run_ir) (in module primeqa.qg.run_qg) (in module primeqa.util.dataloader.file_splitter) make_features() (primeqa.calibration.confidence_scorer.ConfidenceScorer class method) make_training_data() (primeqa.calibration.confidence_scorer.ConfidenceScorer class method) manage_checkpoints_consumed_all_triples() (in module primeqa.ir.dense.colbert_top.colbert.training.utils) manage_checkpoints_with_path_save() (in module primeqa.ir.dense.colbert_top.colbert.training.utils) metric_max_over_ground_truths() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) (in module primeqa.mrc.metrics.squad.evaluate) Metrics (class in primeqa.ir.dense.colbert_top.colbert.evaluation.metrics) metrics_format() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) minimal_answer_span (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) minimal_score (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) mixed_segmentation() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) MixedPrecisionManager (class in primeqa.ir.dense.colbert_top.colbert.utils.amp) MLQA (class in primeqa.mrc.metrics.mlqa.mlqa) model (primeqa.qg.models.qg_model.QGModel property) (primeqa.tableqa.models.tableqa_model.TableQAModel property) model_ (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) model_class_from_config() (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks class method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks class method) model_exists() (primeqa.calibration.confidence_scorer.ConfidenceScorer method) ModelArguments (class in primeqa.calibration.train_confidence_calibrator) (class in primeqa.qg.run_qg) (class in primeqa.tableqa.tapex.utils.argument_utils_for_tapex) ModelForDownstreamTasks (class in primeqa.mrc.models.task_model) module primeqa.boolqa primeqa.boolqa.processors primeqa.boolqa.processors.dataset primeqa.boolqa.processors.dataset.mrc2dataset primeqa.boolqa.run_score_normalizer primeqa.boolqa.score_normalizer primeqa.boolqa.score_normalizer.score_normalizer primeqa.calibration primeqa.calibration.confidence_scorer primeqa.calibration.train_confidence_calibrator primeqa.distillation primeqa.distillation.distillation primeqa.ir primeqa.ir.dense primeqa.ir.dense.colbert_top primeqa.ir.dense.colbert_top.colbert primeqa.ir.dense.colbert_top.colbert.data primeqa.ir.dense.colbert_top.colbert.data.collection primeqa.ir.dense.colbert_top.colbert.data.dataset primeqa.ir.dense.colbert_top.colbert.data.examples primeqa.ir.dense.colbert_top.colbert.data.queries primeqa.ir.dense.colbert_top.colbert.data.ranking primeqa.ir.dense.colbert_top.colbert.distillation primeqa.ir.dense.colbert_top.colbert.distillation.ranking_scorer primeqa.ir.dense.colbert_top.colbert.distillation.scorer primeqa.ir.dense.colbert_top.colbert.evaluation primeqa.ir.dense.colbert_top.colbert.evaluation.load_model primeqa.ir.dense.colbert_top.colbert.evaluation.loaders primeqa.ir.dense.colbert_top.colbert.evaluation.metrics primeqa.ir.dense.colbert_top.colbert.index primeqa.ir.dense.colbert_top.colbert.indexer primeqa.ir.dense.colbert_top.colbert.indexing primeqa.ir.dense.colbert_top.colbert.indexing.codecs primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings_strided primeqa.ir.dense.colbert_top.colbert.indexing.collection_encoder primeqa.ir.dense.colbert_top.colbert.indexing.collection_indexer primeqa.ir.dense.colbert_top.colbert.indexing.index_manager primeqa.ir.dense.colbert_top.colbert.indexing.index_saver primeqa.ir.dense.colbert_top.colbert.indexing.loaders primeqa.ir.dense.colbert_top.colbert.indexing.utils primeqa.ir.dense.colbert_top.colbert.infra primeqa.ir.dense.colbert_top.colbert.infra.config primeqa.ir.dense.colbert_top.colbert.infra.launcher primeqa.ir.dense.colbert_top.colbert.infra.provenance primeqa.ir.dense.colbert_top.colbert.infra.run primeqa.ir.dense.colbert_top.colbert.infra.utilities primeqa.ir.dense.colbert_top.colbert.infra.utilities.minicorpus primeqa.ir.dense.colbert_top.colbert.modeling primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint primeqa.ir.dense.colbert_top.colbert.modeling.colbert primeqa.ir.dense.colbert_top.colbert.modeling.factory primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr primeqa.ir.dense.colbert_top.colbert.modeling.reranker primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra primeqa.ir.dense.colbert_top.colbert.modeling.reranker.tokenizer primeqa.ir.dense.colbert_top.colbert.modeling.tokenization primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_roberta primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_xlmr primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_roberta primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_xlmr primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.utils primeqa.ir.dense.colbert_top.colbert.parameters primeqa.ir.dense.colbert_top.colbert.ranking primeqa.ir.dense.colbert_top.colbert.run_indexer primeqa.ir.dense.colbert_top.colbert.run_searcher primeqa.ir.dense.colbert_top.colbert.run_trainer primeqa.ir.dense.colbert_top.colbert.search primeqa.ir.dense.colbert_top.colbert.search.candidate_generation primeqa.ir.dense.colbert_top.colbert.search.index_loader primeqa.ir.dense.colbert_top.colbert.search.index_storage primeqa.ir.dense.colbert_top.colbert.search.strided_tensor primeqa.ir.dense.colbert_top.colbert.search.strided_tensor_core primeqa.ir.dense.colbert_top.colbert.searcher primeqa.ir.dense.colbert_top.colbert.trainer primeqa.ir.dense.colbert_top.colbert.training primeqa.ir.dense.colbert_top.colbert.training.eager_batcher_v2 primeqa.ir.dense.colbert_top.colbert.training.lazy_batcher primeqa.ir.dense.colbert_top.colbert.training.rerank_batcher primeqa.ir.dense.colbert_top.colbert.training.training primeqa.ir.dense.colbert_top.colbert.training.utils primeqa.ir.dense.colbert_top.colbert.utilities primeqa.ir.dense.colbert_top.colbert.utilities.minicorpus primeqa.ir.dense.colbert_top.colbert.utils primeqa.ir.dense.colbert_top.colbert.utils.amp primeqa.ir.dense.colbert_top.colbert.utils.distributed primeqa.ir.dense.colbert_top.colbert.utils.logging primeqa.ir.dense.colbert_top.colbert.utils.parser primeqa.ir.dense.colbert_top.colbert.utils.runs primeqa.ir.dense.colbert_top.colbert.utils.signals primeqa.ir.dense.colbert_top.colbert.utils.utils primeqa.ir.dense.colbert_top.utility primeqa.ir.dense.colbert_top.utility.evaluate primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers primeqa.ir.dense.colbert_top.utility.preprocess primeqa.ir.dense.colbert_top.utility.preprocess.docs2passages primeqa.ir.dense.colbert_top.utility.rankings primeqa.ir.dense.colbert_top.utility.rankings.dev_subsample primeqa.ir.dense.colbert_top.utility.rankings.merge primeqa.ir.dense.colbert_top.utility.rankings.split_by_offset primeqa.ir.dense.colbert_top.utility.rankings.split_by_queries primeqa.ir.dense.colbert_top.utility.rankings.tune primeqa.ir.dense.colbert_top.utility.supervision primeqa.ir.dense.colbert_top.utility.utils primeqa.ir.dense.colbert_top.utility.utils.qa_loaders primeqa.ir.dense.colbert_top.utility.utils.save_metadata primeqa.ir.dense.dpr_top primeqa.ir.dense.dpr_top.dataloader primeqa.ir.dense.dpr_top.dataloader.distloader_base primeqa.ir.dense.dpr_top.dpr primeqa.ir.dense.dpr_top.dpr.biencoder_gcp primeqa.ir.dense.dpr_top.dpr.biencoder_hypers primeqa.ir.dense.dpr_top.dpr.biencoder_trainer primeqa.ir.dense.dpr_top.dpr.config primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder primeqa.ir.dense.dpr_top.dpr.dpr_util primeqa.ir.dense.dpr_top.dpr.faiss_index primeqa.ir.dense.dpr_top.dpr.index_simple_corpus primeqa.ir.dense.dpr_top.dpr.searcher primeqa.ir.dense.dpr_top.dpr.simple_mmap_dataset primeqa.ir.dense.dpr_top.torch_util primeqa.ir.dense.dpr_top.torch_util.hypers_base primeqa.ir.dense.dpr_top.torch_util.transformer_optimize primeqa.ir.dense.dpr_top.util primeqa.ir.dense.dpr_top.util.args_help primeqa.ir.dense.dpr_top.util.line_corpus primeqa.ir.dense.dpr_top.util.reporting primeqa.ir.run_bm25_retrieval primeqa.ir.run_ir primeqa.ir.sparse primeqa.ir.sparse.bm25_engine primeqa.ir.sparse.config primeqa.ir.sparse.indexer primeqa.ir.sparse.retriever primeqa.ir.sparse.utils primeqa.ir.util primeqa.ir.util.corpus_reader primeqa.mrc primeqa.mrc.data_models primeqa.mrc.data_models.data_collator primeqa.mrc.data_models.eval_prediction_with_processing primeqa.mrc.data_models.model_outputs primeqa.mrc.data_models.model_outputs.extractive primeqa.mrc.data_models.subsample_type primeqa.mrc.data_models.target_type primeqa.mrc.metrics primeqa.mrc.metrics.mlqa primeqa.mrc.metrics.mlqa.mlqa primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1 primeqa.mrc.metrics.nq_f1 primeqa.mrc.metrics.nq_f1.eval_utils primeqa.mrc.metrics.nq_f1.nq_eval primeqa.mrc.metrics.nq_f1.nq_f1 primeqa.mrc.metrics.squad primeqa.mrc.metrics.squad.evaluate primeqa.mrc.metrics.squad.squad primeqa.mrc.metrics.tydi_f1 primeqa.mrc.metrics.tydi_f1.eval_utils primeqa.mrc.metrics.tydi_f1.tydi_eval primeqa.mrc.metrics.tydi_f1.tydi_f1 primeqa.mrc.models primeqa.mrc.models.fid_task_model primeqa.mrc.models.heads primeqa.mrc.models.heads.abstract primeqa.mrc.models.heads.extractive primeqa.mrc.models.heads.generative primeqa.mrc.models.task_model primeqa.mrc.processors primeqa.mrc.processors.postprocessors primeqa.mrc.processors.postprocessors.abstract primeqa.mrc.processors.postprocessors.eli5_fid primeqa.mrc.processors.postprocessors.extractive primeqa.mrc.processors.postprocessors.natural_questions primeqa.mrc.processors.postprocessors.scorers primeqa.mrc.processors.postprocessors.squad primeqa.mrc.processors.preprocessors primeqa.mrc.processors.preprocessors.abstract primeqa.mrc.processors.preprocessors.base primeqa.mrc.processors.preprocessors.eli5_fid primeqa.mrc.processors.preprocessors.mrqa primeqa.mrc.processors.preprocessors.natural_questions primeqa.mrc.processors.preprocessors.squad primeqa.mrc.processors.preprocessors.tydiboolqa_bpes primeqa.mrc.processors.preprocessors.tydiqa primeqa.mrc.processors.preprocessors.tydiqa_google primeqa.mrc.run_mrc_utils primeqa.mrc.trainers primeqa.mrc.trainers.mrc primeqa.mrc.trainers.mrc_mskd primeqa.mrc.trainers.seq2seq_mrc primeqa.qg primeqa.qg.models primeqa.qg.models.hybrid_qg primeqa.qg.models.hybrid_qg.path_sampler primeqa.qg.models.passage_qg primeqa.qg.models.passage_qg.answer_sampler primeqa.qg.models.qg_model primeqa.qg.models.table_qg primeqa.qg.models.table_qg.sql_sampler primeqa.qg.processors primeqa.qg.processors.data_loader primeqa.qg.processors.hybrid_qg primeqa.qg.processors.hybrid_qg.hybridqa_processor primeqa.qg.processors.passage_qg primeqa.qg.processors.passage_qg.qg_processor primeqa.qg.processors.table_qg primeqa.qg.processors.table_qg.sql_processor primeqa.qg.run_qg primeqa.qg.trainers primeqa.qg.trainers.qg_trainer primeqa.qg.utils primeqa.qg.utils.constants primeqa.qg.utils.data_collator primeqa.tableqa primeqa.tableqa.metrics primeqa.tableqa.metrics.answer_accuracy primeqa.tableqa.models primeqa.tableqa.models.tableqa_model primeqa.tableqa.postprocessor primeqa.tableqa.postprocessor.wikisql primeqa.tableqa.preprocessors primeqa.tableqa.preprocessors.dataset primeqa.tableqa.tapex primeqa.tableqa.tapex.metrics primeqa.tableqa.tapex.metrics.tapex_accuracy primeqa.tableqa.tapex.processors primeqa.tableqa.tapex.processors.preprocessors primeqa.tableqa.tapex.utils primeqa.tableqa.tapex.utils.argument_utils_for_tapex primeqa.tableqa.tapex.utils.wikisql_utils_for_tapex primeqa.tableqa.trainers primeqa.tableqa.trainers.tableqa_trainer primeqa.tableqa.utils primeqa.tableqa.utils.data_collator primeqa.util primeqa.util.args_helper primeqa.util.dataloader primeqa.util.dataloader.distloader_base primeqa.util.dataloader.distloader_seq_pair primeqa.util.dataloader.file_splitter primeqa.util.file_utils primeqa.util.metrics primeqa.util.reporting primeqa.util.transformers_utils primeqa.util.transformers_utils.hypers_base primeqa.util.transformers_utils.model_utils primeqa.util.transformers_utils.optimizer_utils primeqa.util.transformers_utils.torch_utils modules() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) move_to_end() (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) MRCSeq2SeqTrainer (class in primeqa.mrc.trainers.seq2seq_mrc) MRCTrainer (class in primeqa.mrc.trainers.mrc) MRQAPreprocessor (class in primeqa.mrc.processors.preprocessors.mrqa) MSKD_MRCTrainer (class in primeqa.mrc.trainers.mrc_mskd) multiclass_score_metrics() (in module primeqa.util.metrics) MultiFileLoader (class in primeqa.ir.dense.dpr_top.dataloader.distloader_base) (class in primeqa.util.dataloader.distloader_base) N name_value_list() (in module primeqa.ir.dense.dpr_top.util.args_help) (in module primeqa.util.args_helper) named_buffers() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) named_children() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) named_modules() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) named_parameters() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) NaturalQuestionsPostProcessor (class in primeqa.mrc.processors.postprocessors.natural_questions) NaturalQuestionsPreProcessor (class in primeqa.mrc.processors.preprocessors.natural_questions) nonnull_span_equal() (in module primeqa.mrc.metrics.nq_f1.eval_utils) (in module primeqa.mrc.metrics.tydi_f1.eval_utils) normalize_answer() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) (in module primeqa.mrc.metrics.squad.evaluate) np2str() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) NQF1 (class in primeqa.mrc.metrics.nq_f1.nq_f1) NQLabel (class in primeqa.mrc.metrics.nq_f1.eval_utils) NQSpan (class in primeqa.mrc.metrics.nq_f1.eval_utils) NullContextManager (class in primeqa.ir.dense.colbert_top.colbert.utils.utils) num_examples() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) num_parameters() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) numericize() (in module primeqa.ir.dense.colbert_top.colbert.data.ranking) O object_reference() (in module primeqa.calibration.train_confidence_calibrator) (in module primeqa.mrc.run_mrc_utils) optimize_ivf() (in module primeqa.ir.dense.colbert_top.colbert.indexing.utils) Options (class in primeqa.ir.dense.dpr_top.dpr.index_simple_corpus) (class in primeqa.ir.dense.dpr_top.dpr.searcher) (class in primeqa.util.dataloader.file_splitter) P parameters() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) parse_arguments() (in module primeqa.boolqa.run_score_normalizer) parse_json_as_predictions() (in module primeqa.mrc.metrics.nq_f1.eval_utils) parse_json_as_topk_predictions() (in module primeqa.mrc.metrics.nq_f1.eval_utils) Passage (class in primeqa.ir.util.corpus_reader) passage_answer_index (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) passage_score (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) passage_span (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) PathSampler (class in primeqa.qg.models.hybrid_qg.path_sampler) plaintext (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) pop() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) pop_callback() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) popitem() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) post_init() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) postprocess_prediction() (primeqa.tableqa.postprocessor.wikisql.WikiSQLPostprocessor method) postprocessor (primeqa.calibration.train_confidence_calibrator.TaskArguments attribute) predict() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.models.tableqa_model.TableQAModel method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) predict_scores() (primeqa.calibration.confidence_scorer.ConfidenceScorer method) prediction_loop() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) prediction_step() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) predictions (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing attribute) prepare_examples_as_references() (primeqa.mrc.processors.postprocessors.abstract.AbstractPostProcessor method) (primeqa.mrc.processors.postprocessors.eli5_fid.ELI5FiDPostProcessor method) (primeqa.mrc.processors.postprocessors.extractive.ExtractivePostProcessor method) (primeqa.mrc.processors.postprocessors.natural_questions.NaturalQuestionsPostProcessor method) (primeqa.mrc.processors.postprocessors.squad.SQUADPostProcessor method) prepare_inputs_for_generation() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) preprocess_data() (primeqa.qg.processors.table_qg.sql_processor.SqlProcessor static method) preprocess_hybridqa_data() (primeqa.qg.processors.hybrid_qg.hybridqa_processor.HybridQAProcessor method) preprocessor (primeqa.calibration.train_confidence_calibrator.TaskArguments attribute) pretty_print() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) primeqa.boolqa module primeqa.boolqa.processors module primeqa.boolqa.processors.dataset module primeqa.boolqa.processors.dataset.mrc2dataset module primeqa.boolqa.run_score_normalizer module primeqa.boolqa.score_normalizer module primeqa.boolqa.score_normalizer.score_normalizer module primeqa.calibration module primeqa.calibration.confidence_scorer module primeqa.calibration.train_confidence_calibrator module primeqa.distillation module primeqa.distillation.distillation module primeqa.ir module primeqa.ir.dense module primeqa.ir.dense.colbert_top module primeqa.ir.dense.colbert_top.colbert module primeqa.ir.dense.colbert_top.colbert.data module primeqa.ir.dense.colbert_top.colbert.data.collection module primeqa.ir.dense.colbert_top.colbert.data.dataset module primeqa.ir.dense.colbert_top.colbert.data.examples module primeqa.ir.dense.colbert_top.colbert.data.queries module primeqa.ir.dense.colbert_top.colbert.data.ranking module primeqa.ir.dense.colbert_top.colbert.distillation module primeqa.ir.dense.colbert_top.colbert.distillation.ranking_scorer module primeqa.ir.dense.colbert_top.colbert.distillation.scorer module primeqa.ir.dense.colbert_top.colbert.evaluation module primeqa.ir.dense.colbert_top.colbert.evaluation.load_model module primeqa.ir.dense.colbert_top.colbert.evaluation.loaders module primeqa.ir.dense.colbert_top.colbert.evaluation.metrics module primeqa.ir.dense.colbert_top.colbert.index module primeqa.ir.dense.colbert_top.colbert.indexer module primeqa.ir.dense.colbert_top.colbert.indexing module primeqa.ir.dense.colbert_top.colbert.indexing.codecs module primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual module primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings module primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings_strided module primeqa.ir.dense.colbert_top.colbert.indexing.collection_encoder module primeqa.ir.dense.colbert_top.colbert.indexing.collection_indexer module primeqa.ir.dense.colbert_top.colbert.indexing.index_manager module primeqa.ir.dense.colbert_top.colbert.indexing.index_saver module primeqa.ir.dense.colbert_top.colbert.indexing.loaders module primeqa.ir.dense.colbert_top.colbert.indexing.utils module primeqa.ir.dense.colbert_top.colbert.infra module primeqa.ir.dense.colbert_top.colbert.infra.config module primeqa.ir.dense.colbert_top.colbert.infra.launcher module primeqa.ir.dense.colbert_top.colbert.infra.provenance module primeqa.ir.dense.colbert_top.colbert.infra.run module primeqa.ir.dense.colbert_top.colbert.infra.utilities module primeqa.ir.dense.colbert_top.colbert.infra.utilities.minicorpus module primeqa.ir.dense.colbert_top.colbert.modeling module primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert module primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint module primeqa.ir.dense.colbert_top.colbert.modeling.colbert module primeqa.ir.dense.colbert_top.colbert.modeling.factory module primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert module primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta module primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr module primeqa.ir.dense.colbert_top.colbert.modeling.reranker module primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra module primeqa.ir.dense.colbert_top.colbert.modeling.reranker.tokenizer module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_roberta module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.doc_tokenization_xlmr module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_roberta module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_xlmr module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.utils module primeqa.ir.dense.colbert_top.colbert.parameters module primeqa.ir.dense.colbert_top.colbert.ranking module primeqa.ir.dense.colbert_top.colbert.run_indexer module primeqa.ir.dense.colbert_top.colbert.run_searcher module primeqa.ir.dense.colbert_top.colbert.run_trainer module primeqa.ir.dense.colbert_top.colbert.search module primeqa.ir.dense.colbert_top.colbert.search.candidate_generation module primeqa.ir.dense.colbert_top.colbert.search.index_loader module primeqa.ir.dense.colbert_top.colbert.search.index_storage module primeqa.ir.dense.colbert_top.colbert.search.strided_tensor module primeqa.ir.dense.colbert_top.colbert.search.strided_tensor_core module primeqa.ir.dense.colbert_top.colbert.searcher module primeqa.ir.dense.colbert_top.colbert.trainer module primeqa.ir.dense.colbert_top.colbert.training module primeqa.ir.dense.colbert_top.colbert.training.eager_batcher_v2 module primeqa.ir.dense.colbert_top.colbert.training.lazy_batcher module primeqa.ir.dense.colbert_top.colbert.training.rerank_batcher module primeqa.ir.dense.colbert_top.colbert.training.training module primeqa.ir.dense.colbert_top.colbert.training.utils module primeqa.ir.dense.colbert_top.colbert.utilities module primeqa.ir.dense.colbert_top.colbert.utilities.minicorpus module primeqa.ir.dense.colbert_top.colbert.utils module primeqa.ir.dense.colbert_top.colbert.utils.amp module primeqa.ir.dense.colbert_top.colbert.utils.distributed module primeqa.ir.dense.colbert_top.colbert.utils.logging module primeqa.ir.dense.colbert_top.colbert.utils.parser module primeqa.ir.dense.colbert_top.colbert.utils.runs module primeqa.ir.dense.colbert_top.colbert.utils.signals module primeqa.ir.dense.colbert_top.colbert.utils.utils module primeqa.ir.dense.colbert_top.utility module primeqa.ir.dense.colbert_top.utility.evaluate module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers module primeqa.ir.dense.colbert_top.utility.preprocess module primeqa.ir.dense.colbert_top.utility.preprocess.docs2passages module primeqa.ir.dense.colbert_top.utility.rankings module primeqa.ir.dense.colbert_top.utility.rankings.dev_subsample module primeqa.ir.dense.colbert_top.utility.rankings.merge module primeqa.ir.dense.colbert_top.utility.rankings.split_by_offset module primeqa.ir.dense.colbert_top.utility.rankings.split_by_queries module primeqa.ir.dense.colbert_top.utility.rankings.tune module primeqa.ir.dense.colbert_top.utility.supervision module primeqa.ir.dense.colbert_top.utility.utils module primeqa.ir.dense.colbert_top.utility.utils.qa_loaders module primeqa.ir.dense.colbert_top.utility.utils.save_metadata module primeqa.ir.dense.dpr_top module primeqa.ir.dense.dpr_top.dataloader module primeqa.ir.dense.dpr_top.dataloader.distloader_base module primeqa.ir.dense.dpr_top.dpr module primeqa.ir.dense.dpr_top.dpr.biencoder_gcp module primeqa.ir.dense.dpr_top.dpr.biencoder_hypers module primeqa.ir.dense.dpr_top.dpr.biencoder_trainer module primeqa.ir.dense.dpr_top.dpr.config module primeqa.ir.dense.dpr_top.dpr.dataloader_biencoder module primeqa.ir.dense.dpr_top.dpr.dpr_util module primeqa.ir.dense.dpr_top.dpr.faiss_index module primeqa.ir.dense.dpr_top.dpr.index_simple_corpus module primeqa.ir.dense.dpr_top.dpr.searcher module primeqa.ir.dense.dpr_top.dpr.simple_mmap_dataset module primeqa.ir.dense.dpr_top.torch_util module primeqa.ir.dense.dpr_top.torch_util.hypers_base module primeqa.ir.dense.dpr_top.torch_util.transformer_optimize module primeqa.ir.dense.dpr_top.util module primeqa.ir.dense.dpr_top.util.args_help module primeqa.ir.dense.dpr_top.util.line_corpus module primeqa.ir.dense.dpr_top.util.reporting module primeqa.ir.run_bm25_retrieval module primeqa.ir.run_ir module primeqa.ir.sparse module primeqa.ir.sparse.bm25_engine module primeqa.ir.sparse.config module primeqa.ir.sparse.indexer module primeqa.ir.sparse.retriever module primeqa.ir.sparse.utils module primeqa.ir.util module primeqa.ir.util.corpus_reader module primeqa.mrc module primeqa.mrc.data_models module primeqa.mrc.data_models.data_collator module primeqa.mrc.data_models.eval_prediction_with_processing module primeqa.mrc.data_models.model_outputs module primeqa.mrc.data_models.model_outputs.extractive module primeqa.mrc.data_models.subsample_type module primeqa.mrc.data_models.target_type module primeqa.mrc.metrics module primeqa.mrc.metrics.mlqa module primeqa.mrc.metrics.mlqa.mlqa module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1 module primeqa.mrc.metrics.nq_f1 module primeqa.mrc.metrics.nq_f1.eval_utils module primeqa.mrc.metrics.nq_f1.nq_eval module primeqa.mrc.metrics.nq_f1.nq_f1 module primeqa.mrc.metrics.squad module primeqa.mrc.metrics.squad.evaluate module primeqa.mrc.metrics.squad.squad module primeqa.mrc.metrics.tydi_f1 module primeqa.mrc.metrics.tydi_f1.eval_utils module primeqa.mrc.metrics.tydi_f1.tydi_eval module primeqa.mrc.metrics.tydi_f1.tydi_f1 module primeqa.mrc.models module primeqa.mrc.models.fid_task_model module primeqa.mrc.models.heads module primeqa.mrc.models.heads.abstract module primeqa.mrc.models.heads.extractive module primeqa.mrc.models.heads.generative module primeqa.mrc.models.task_model module primeqa.mrc.processors module primeqa.mrc.processors.postprocessors module primeqa.mrc.processors.postprocessors.abstract module primeqa.mrc.processors.postprocessors.eli5_fid module primeqa.mrc.processors.postprocessors.extractive module primeqa.mrc.processors.postprocessors.natural_questions module primeqa.mrc.processors.postprocessors.scorers module primeqa.mrc.processors.postprocessors.squad module primeqa.mrc.processors.preprocessors module primeqa.mrc.processors.preprocessors.abstract module primeqa.mrc.processors.preprocessors.base module primeqa.mrc.processors.preprocessors.eli5_fid module primeqa.mrc.processors.preprocessors.mrqa module primeqa.mrc.processors.preprocessors.natural_questions module primeqa.mrc.processors.preprocessors.squad module primeqa.mrc.processors.preprocessors.tydiboolqa_bpes module primeqa.mrc.processors.preprocessors.tydiqa module primeqa.mrc.processors.preprocessors.tydiqa_google module primeqa.mrc.run_mrc_utils module primeqa.mrc.trainers module primeqa.mrc.trainers.mrc module primeqa.mrc.trainers.mrc_mskd module primeqa.mrc.trainers.seq2seq_mrc module primeqa.qg module primeqa.qg.models module primeqa.qg.models.hybrid_qg module primeqa.qg.models.hybrid_qg.path_sampler module primeqa.qg.models.passage_qg module primeqa.qg.models.passage_qg.answer_sampler module primeqa.qg.models.qg_model module primeqa.qg.models.table_qg module primeqa.qg.models.table_qg.sql_sampler module primeqa.qg.processors module primeqa.qg.processors.data_loader module primeqa.qg.processors.hybrid_qg module primeqa.qg.processors.hybrid_qg.hybridqa_processor module primeqa.qg.processors.passage_qg module primeqa.qg.processors.passage_qg.qg_processor module primeqa.qg.processors.table_qg module primeqa.qg.processors.table_qg.sql_processor module primeqa.qg.run_qg module primeqa.qg.trainers module primeqa.qg.trainers.qg_trainer module primeqa.qg.utils module primeqa.qg.utils.constants module primeqa.qg.utils.data_collator module primeqa.tableqa module primeqa.tableqa.metrics module primeqa.tableqa.metrics.answer_accuracy module primeqa.tableqa.models module primeqa.tableqa.models.tableqa_model module primeqa.tableqa.postprocessor module primeqa.tableqa.postprocessor.wikisql module primeqa.tableqa.preprocessors module primeqa.tableqa.preprocessors.dataset module primeqa.tableqa.tapex module primeqa.tableqa.tapex.metrics module primeqa.tableqa.tapex.metrics.tapex_accuracy module primeqa.tableqa.tapex.processors module primeqa.tableqa.tapex.processors.preprocessors module primeqa.tableqa.tapex.utils module primeqa.tableqa.tapex.utils.argument_utils_for_tapex module primeqa.tableqa.tapex.utils.wikisql_utils_for_tapex module primeqa.tableqa.trainers module primeqa.tableqa.trainers.tableqa_trainer module primeqa.tableqa.utils module primeqa.tableqa.utils.data_collator module primeqa.util module primeqa.util.args_helper module primeqa.util.dataloader module primeqa.util.dataloader.distloader_base module primeqa.util.dataloader.distloader_seq_pair module primeqa.util.dataloader.file_splitter module primeqa.util.file_utils module primeqa.util.metrics module primeqa.util.reporting module primeqa.util.transformers_utils module primeqa.util.transformers_utils.hypers_base module primeqa.util.transformers_utils.model_utils module primeqa.util.transformers_utils.optimizer_utils module primeqa.util.transformers_utils.torch_utils module print_memory_stats() (in module primeqa.ir.dense.colbert_top.colbert.infra.launcher) print_message() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) print_progress() (in module primeqa.ir.dense.colbert_top.colbert.training.utils) print_r_at_p_table() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) print_torch_extension_error_message() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) process() (primeqa.mrc.processors.postprocessors.abstract.AbstractPostProcessor method) (primeqa.mrc.processors.postprocessors.eli5_fid.ELI5FiDPostProcessor method) (primeqa.mrc.processors.postprocessors.extractive.ExtractivePostProcessor method) (primeqa.mrc.processors.postprocessors.natural_questions.NaturalQuestionsPostProcessor method) (primeqa.mrc.processors.postprocessors.squad.SQUADPostProcessor method) process_eval() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) process_grouped_by_first_item() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) process_page() (in module primeqa.ir.dense.colbert_top.utility.preprocess.docs2passages) process_raw_datasets() (in module primeqa.mrc.run_mrc_utils) process_references_and_predictions() (primeqa.mrc.processors.postprocessors.abstract.AbstractPostProcessor method) (primeqa.mrc.processors.postprocessors.eli5_fid.ELI5FiDPostProcessor method) (primeqa.mrc.processors.postprocessors.extractive.ExtractivePostProcessor method) (primeqa.mrc.processors.postprocessors.natural_questions.NaturalQuestionsPostProcessor method) (primeqa.mrc.processors.postprocessors.squad.SQUADPostProcessor method) process_train() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) ProcessArguments (class in primeqa.ir.run_ir) processed_predictions (primeqa.mrc.data_models.eval_prediction_with_processing.EvalPredictionWithProcessing attribute) Provenance (class in primeqa.ir.dense.colbert_top.colbert.infra.provenance) prune_hallucinations() (primeqa.qg.models.qg_model.QGModel method) prune_heads() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) push_to_hub() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) PyseriniIndexer (class in primeqa.ir.sparse.indexer) PyseriniRetriever (class in primeqa.ir.sparse.retriever) Q QGDataLoader (class in primeqa.qg.processors.data_loader) QGModel (class in primeqa.qg.models.qg_model) QGProcessor (class in primeqa.qg.processors.passage_qg.qg_processor) QGSpecialTokens (class in primeqa.qg.utils.constants) QGTrainer (class in primeqa.qg.trainers.qg_trainer) Queries (class in primeqa.ir.dense.colbert_top.colbert.data.queries) queries_to_vectors() (in module primeqa.ir.dense.dpr_top.dpr.dpr_util) QueryTokenizer (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization) QueryTokenizerRoberta (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_roberta) QueryTokenizerXLMR (class in primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.query_tokenization_xlmr) question_text (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) R Ranking (class in primeqa.ir.dense.colbert_top.colbert.data.ranking) RankingScorer (class in primeqa.ir.dense.colbert_top.colbert.distillation.ranking_scorer) read_annotation() (in module primeqa.mrc.metrics.nq_f1.eval_utils) read_annotation_from_one_split() (in module primeqa.mrc.metrics.nq_f1.eval_utils) read_lines() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) read_open() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) read_prediction_json_from_file() (in module primeqa.mrc.metrics.nq_f1.eval_utils) read_records() (in module primeqa.ir.dense.dpr_top.util.line_corpus) readable_sql() (primeqa.qg.models.table_qg.sql_sampler.SimpleSqlSampler method) reduce() (in module primeqa.util.transformers_utils.torch_utils) reference_prediction_overlap() (primeqa.calibration.confidence_scorer.ConfidenceScorer class method) register_backward_hook() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_buffer() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_for_auto_class() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT class method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta class method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR class method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker class method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks class method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks class method) register_forward_hook() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_forward_pre_hook() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_full_backward_hook() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_module() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) register_parameter() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) remove_callback() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) remove_first_and_last_quote() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) Reporting (class in primeqa.ir.dense.dpr_top.util.reporting) (class in primeqa.util.reporting) requires_grad_() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) RerankBatcher (class in primeqa.ir.dense.colbert_top.colbert.training.rerank_batcher) RerankerTokenizer (class in primeqa.ir.dense.colbert_top.colbert.modeling.reranker.tokenizer) reset_memory_hooks_state() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) ResidualCodec (class in primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual) ResidualEmbeddings (class in primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings) ResidualEmbeddingsStrided (class in primeqa.ir.dense.colbert_top.colbert.indexing.codecs.residual_embeddings_strided) resize_token_embeddings() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) retrieve() (primeqa.ir.sparse.retriever.PyseriniRetriever method) retrieve_wikisql_query_answer_tapas() (in module primeqa.tableqa.tapex.utils.wikisql_utils_for_tapex) Run (class in primeqa.ir.dense.colbert_top.colbert.infra.run) run_search() (in module primeqa.ir.run_bm25_retrieval) RunConfig (class in primeqa.ir.dense.colbert_top.colbert.infra.config) S safe_average() (in module primeqa.mrc.metrics.tydi_f1.eval_utils) safe_divide() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.eval_utils) sample() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) sample_minicorpus() (in module primeqa.ir.dense.colbert_top.colbert.infra.utilities.minicorpus) (in module primeqa.ir.dense.colbert_top.colbert.utilities.minicorpus) sample_sql() (primeqa.qg.models.table_qg.sql_sampler.SimpleSqlSampler method) save_checkpoint() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) save_extended_model() (in module primeqa.util.transformers_utils.model_utils) save_metadata() (in module primeqa.ir.dense.colbert_top.utility.utils.save_metadata) save_metrics() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) save_model() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) save_pretrained() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) save_ranking() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) save_state() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) save_transformer() (in module primeqa.util.transformers_utils.model_utils) score_answers() (in module primeqa.mrc.metrics.nq_f1.nq_eval) (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) score_long_answer() (in module primeqa.mrc.metrics.nq_f1.nq_eval) score_metrics() (in module primeqa.util.metrics) score_minimal_answer() (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) score_passage_answer() (in module primeqa.mrc.metrics.tydi_f1.tydi_eval) score_pids() (primeqa.ir.dense.colbert_top.colbert.search.index_storage.IndexScorer method) score_short_answer() (in module primeqa.mrc.metrics.nq_f1.nq_eval) ScoreNormalizer (class in primeqa.boolqa.score_normalizer.score_normalizer) Scorer (class in primeqa.ir.dense.colbert_top.colbert.distillation.scorer) SearchArguments (class in primeqa.ir.sparse.config) Searcher (class in primeqa.ir.dense.colbert_top.colbert.searcher) sentence_to_inputs() (in module primeqa.util.dataloader.distloader_base) SeqPairBatches (class in primeqa.util.dataloader.distloader_seq_pair) SeqPairInst (class in primeqa.util.dataloader.distloader_seq_pair) SeqPairLoader (class in primeqa.util.dataloader.distloader_seq_pair) set_bert_grad() (in module primeqa.ir.dense.colbert_top.colbert.training.training) set_extra_state() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) set_gradient_accumulation_steps() (primeqa.ir.dense.dpr_top.dpr.biencoder_hypers.BiEncoderHypers method) (primeqa.ir.dense.dpr_top.dpr.biencoder_trainer.BiEncoderTrainArgs method) (primeqa.ir.dense.dpr_top.dpr.dpr_util.DPROptions method) (primeqa.ir.dense.dpr_top.dpr.searcher.Options method) (primeqa.ir.dense.dpr_top.torch_util.hypers_base.HypersBase method) (primeqa.util.transformers_utils.hypers_base.HypersBase method) set_input_embeddings() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) set_task_head() (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) setdefault() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) setup_new_process() (in module primeqa.ir.dense.colbert_top.colbert.infra.launcher) share_memory() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) short_answer_span_list (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) short_score (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) shuffled_writer() (in module primeqa.util.file_utils) ShuffledWriter (class in primeqa.ir.dense.dpr_top.util.line_corpus) (class in primeqa.util.file_utils) simple_accuracy() (in module primeqa.util.metrics) SimpleSqlSampler (class in primeqa.qg.models.table_qg.sql_sampler) Span (class in primeqa.mrc.metrics.tydi_f1.eval_utils) span_set_equal() (in module primeqa.mrc.metrics.nq_f1.eval_utils) sql_execution() (primeqa.qg.models.table_qg.sql_sampler.SimpleSqlSampler method) SqlOperants (class in primeqa.qg.utils.constants) SqlProcessor (class in primeqa.qg.processors.table_qg.sql_processor) SQUAD (class in primeqa.mrc.metrics.squad.squad) SQUADPostProcessor (class in primeqa.mrc.processors.postprocessors.squad) SQUADPreprocessor (class in primeqa.mrc.processors.preprocessors.squad) standard_json_mapper() (in module primeqa.util.dataloader.distloader_seq_pair) state_dict() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) stdout_or_file_open() (in module primeqa.util.file_utils) step_loss() (primeqa.ir.dense.dpr_top.torch_util.transformer_optimize.TransformerOptimize method) str2np() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) StridedTensor (class in primeqa.ir.dense.colbert_top.colbert.search.strided_tensor) StridedTensorCore (class in primeqa.ir.dense.colbert_top.colbert.search.strided_tensor_core) subsample_features() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) SubsampleType (class in primeqa.mrc.data_models.subsample_type) SupportedSpanScorers (class in primeqa.mrc.processors.postprocessors.scorers) T T2TDataCollator (class in primeqa.qg.utils.data_collator) TableQADataset (class in primeqa.tableqa.preprocessors.dataset) TableQAModel (class in primeqa.tableqa.models.tableqa_model) TableQATrainer (class in primeqa.tableqa.trainers.tableqa_trainer) TapasCollator (class in primeqa.tableqa.utils.data_collator) TapexAccuracy (class in primeqa.tableqa.tapex.metrics.tapex_accuracy) TargetType (class in primeqa.mrc.data_models.target_type) task_head (primeqa.mrc.models.task_model.ModelForDownstreamTasks property) TaskArguments (class in primeqa.calibration.train_confidence_calibrator) tensorize_triples() (in module primeqa.ir.dense.colbert_top.colbert.modeling.tokenization.utils) tie_weights() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) time_str() (in module primeqa.ir.dense.dpr_top.util.reporting) (in module primeqa.util.reporting) timestamp() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) to() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) to_dict() (primeqa.ir.util.corpus_reader.Passage method) to_empty() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) to_tensor() (in module primeqa.util.transformers_utils.torch_utils) to_tuple() (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) tokenize_all_answers() (in module primeqa.ir.dense.colbert_top.utility.evaluate.annotate_EM_helpers) tokenize_queries() (in module primeqa.ir.dense.dpr_top.dpr.dpr_util) tokenizer (primeqa.qg.models.qg_model.QGModel property) (primeqa.tableqa.models.tableqa_model.TableQAModel property) tolist() (primeqa.ir.dense.colbert_top.colbert.data.examples.Examples method) torch_load_dnn() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) train() (in module primeqa.ir.dense.colbert_top.colbert.training.training) (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.colbert_top.colbert.trainer.Trainer method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) Trainer (class in primeqa.ir.dense.colbert_top.colbert.trainer) training_step() (primeqa.mrc.trainers.mrc.MRCTrainer method) (primeqa.mrc.trainers.mrc_mskd.MSKD_MRCTrainer method) (primeqa.mrc.trainers.seq2seq_mrc.MRCSeq2SeqTrainer method) (primeqa.qg.trainers.qg_trainer.QGTrainer method) (primeqa.tableqa.trainers.tableqa_trainer.TableQATrainer method) TransformerOptimize (class in primeqa.ir.dense.dpr_top.torch_util.transformer_optimize) (class in primeqa.util.transformers_utils.optimizer_utils) TyDiBoolQAPreprocessor (class in primeqa.mrc.processors.preprocessors.tydiboolqa_bpes) TyDiF1 (class in primeqa.mrc.metrics.tydi_f1.tydi_f1) TyDiLabel (class in primeqa.mrc.metrics.tydi_f1.eval_utils) TyDiQAGooglePreprocessor (class in primeqa.mrc.processors.preprocessors.tydiqa_google) TyDiQAPreprocessor (class in primeqa.mrc.processors.preprocessors.tydiqa) type() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) U unpack_span_answer() (in module primeqa.boolqa.processors.dataset.mrc2dataset) unpack_target_type() (in module primeqa.boolqa.processors.dataset.mrc2dataset) update() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) update_keys_to_ignore() (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) V validate_schema() (primeqa.mrc.processors.preprocessors.abstract.AbstractPreProcessor method) (primeqa.mrc.processors.preprocessors.base.BasePreProcessor method) (primeqa.mrc.processors.preprocessors.eli5_fid.ELI5FiDPreprocessor method) (primeqa.mrc.processors.preprocessors.mrqa.MRQAPreprocessor method) (primeqa.mrc.processors.preprocessors.natural_questions.NaturalQuestionsPreProcessor method) (primeqa.mrc.processors.preprocessors.squad.SQUADPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiboolqa_bpes.TyDiBoolQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa.TyDiQAPreprocessor method) (primeqa.mrc.processors.preprocessors.tydiqa_google.TyDiQAGooglePreprocessor method) values() (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict method) (primeqa.ir.dense.colbert_top.colbert.utils.utils.dotdict_lax method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAModelOutput method) (primeqa.mrc.data_models.model_outputs.extractive.ExtractiveQAWithConfidenceModelOutput method) W whitespace_tokenize() (in module primeqa.mrc.metrics.mlqa.mlqa_evaluation_v1) WikiSQLPostprocessor (class in primeqa.tableqa.postprocessor.wikisql) write_colbert_ranking_tsv() (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.sparse.utils) write_corpus_tsv() (primeqa.ir.util.corpus_reader.DocumentCollection method) write_open() (in module primeqa.ir.dense.dpr_top.util.line_corpus) (in module primeqa.util.file_utils) write_search_results() (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.sparse.utils) write_xorqa_json() (in module primeqa.ir.run_bm25_retrieval) (in module primeqa.ir.sparse.utils) X xpu() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) Y yes_no_answer (primeqa.mrc.metrics.nq_f1.eval_utils.NQLabel attribute) (primeqa.mrc.metrics.tydi_f1.eval_utils.TyDiLabel attribute) Z zero_grad() (primeqa.ir.dense.colbert_top.colbert.modeling.base_colbert.BaseColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.checkpoint.Checkpoint method) (primeqa.ir.dense.colbert_top.colbert.modeling.colbert.ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert.HF_ColBERT method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_roberta.HF_ColBERT_Roberta method) (primeqa.ir.dense.colbert_top.colbert.modeling.hf_colbert_xlmr.HF_ColBERT_XLMR method) (primeqa.ir.dense.colbert_top.colbert.modeling.reranker.electra.ElectraReranker method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.BiEncoder method) (primeqa.ir.dense.dpr_top.dpr.biencoder_gcp.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.EncoderWrapper method) (primeqa.mrc.models.fid_task_model.FiDModelForDownstreamTasks method) (primeqa.mrc.models.heads.abstract.AbstractTaskHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAHead method) (primeqa.mrc.models.heads.extractive.ExtractiveQAWithConfidenceHead method) (primeqa.mrc.models.task_model.ModelForDownstreamTasks method) zip_first() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils) zipstar() (in module primeqa.ir.dense.colbert_top.colbert.utils.utils)