Index

_

_cosine_similarity() (kitanaqa.augment.generators.BaseGenerator method)

A

adv_evaluate() (kitanaqa.trainer.train.Trainer method) alpha (kitanaqa.trainer.arguments.ModelArguments attribute) alpha_final (kitanaqa.trainer.arguments.ModelArguments attribute) alpha_schedule (kitanaqa.trainer.arguments.ModelArguments attribute) alum_get_dev_examples() (kitanaqa.trainer.alum_squad_processor.AlumSquadProcessor method)

alum_squad_convert_examples_to_features() (in module kitanaqa.trainer.alum_squad_processor) AlumSquadProcessor (class in kitanaqa.trainer.alum_squad_processor) AlumSquadV1Processor (class in kitanaqa.trainer.alum_squad_processor) AlumSquadV2Processor (class in kitanaqa.trainer.alum_squad_processor) aug_file_path (kitanaqa.trainer.arguments.ModelArguments attribute)

B

BaseGenerator (class in kitanaqa.augment.generators)

build_flow() (in module kitanaqa.trainer.utils)

C

cache_dir (kitanaqa.trainer.arguments.ModelArguments attribute)

custom_scheduler() (in module kitanaqa.trainer.custom_schedulers)

D

data_dir (kitanaqa.trainer.arguments.ModelArguments attribute) default_logdir() (in module kitanaqa.trainer.arguments) dev_file (kitanaqa.trainer.alum_squad_processor.AlumSquadProcessor attribute) (kitanaqa.trainer.alum_squad_processor.AlumSquadV1Processor attribute) (kitanaqa.trainer.alum_squad_processor.AlumSquadV2Processor attribute) do_adv_eval (kitanaqa.trainer.arguments.ModelArguments attribute)

do_alum (kitanaqa.trainer.arguments.ModelArguments attribute) do_aug (kitanaqa.trainer.arguments.ModelArguments attribute) do_lower_case (kitanaqa.trainer.arguments.ModelArguments attribute) doc_stride (kitanaqa.trainer.arguments.ModelArguments attribute) drop_terms() (kitanaqa.augment.term_replacement.DropTerms method) DropTerms (class in kitanaqa.augment.term_replacement)

E

eps (kitanaqa.trainer.arguments.ModelArguments attribute) eta (kitanaqa.trainer.arguments.ModelArguments attribute)

eval_all_checkpoints (kitanaqa.trainer.arguments.ModelArguments attribute) evaluate() (kitanaqa.trainer.train.Trainer method), [1]

F

format_squad() (in module kitanaqa.augment.augment_squad)

freeze_embeds (kitanaqa.trainer.arguments.ModelArguments attribute)

G

generate() (kitanaqa.augment.augment_squad.SQuADDataset method) (kitanaqa.augment.generators.MisspReplace method) (kitanaqa.augment.generators.MLMSynonymReplace method) (kitanaqa.augment.generators.SynonymReplace method)

get_custom_exp() (in module kitanaqa.trainer.custom_schedulers) get_custom_linear() (in module kitanaqa.trainer.custom_schedulers) get_scores() (in module kitanaqa.augment.term_replacement)

K

K (kitanaqa.trainer.arguments.ModelArguments attribute) kitanaqa.augment module kitanaqa.augment.augment_squad module kitanaqa.augment.generators module kitanaqa.augment.term_replacement module kitanaqa.trainer module

kitanaqa.trainer.alum_squad_processor module kitanaqa.trainer.arguments module kitanaqa.trainer.custom_schedulers module kitanaqa.trainer.train module kitanaqa.trainer.utils module

L

load_and_cache_examples() (in module kitanaqa.trainer.utils)

log() (kitanaqa.trainer.train.Trainer method), [1]

M

max_answer_length (kitanaqa.trainer.arguments.ModelArguments attribute) max_query_length (kitanaqa.trainer.arguments.ModelArguments attribute) max_seq_length (kitanaqa.trainer.arguments.ModelArguments attribute) MisspReplace (class in kitanaqa.augment.generators) MLMSynonymReplace (class in kitanaqa.augment.generators) model_args (kitanaqa.trainer.train.Trainer attribute) model_name_or_path (kitanaqa.trainer.arguments.ModelArguments attribute) model_type (kitanaqa.trainer.arguments.ModelArguments attribute) ModelArguments (class in kitanaqa.trainer.arguments) module kitanaqa.augment kitanaqa.augment.augment_squad kitanaqa.augment.generators kitanaqa.augment.term_replacement kitanaqa.trainer kitanaqa.trainer.alum_squad_processor kitanaqa.trainer.arguments kitanaqa.trainer.custom_schedulers kitanaqa.trainer.train kitanaqa.trainer.utils

N

n_best_size (kitanaqa.trainer.arguments.ModelArguments attribute)

null_score_diff_threshold (kitanaqa.trainer.arguments.ModelArguments attribute)

O

overwrite_cache (kitanaqa.trainer.arguments.ModelArguments attribute)

P

post_to_slack() (in module kitanaqa.trainer.utils)

predict_file_path (kitanaqa.trainer.arguments.ModelArguments attribute)

R

repeat_terms() (kitanaqa.augment.term_replacement.RepeatTerms method) RepeatTerms (class in kitanaqa.augment.term_replacement)

replace_terms() (kitanaqa.augment.term_replacement.ReplaceTerms method) ReplaceTerms (class in kitanaqa.augment.term_replacement)

S

setup_comet() (kitanaqa.trainer.train.Trainer method) sigma (kitanaqa.trainer.arguments.ModelArguments attribute)

SQuADDataset (class in kitanaqa.augment.augment_squad) SynonymReplace (class in kitanaqa.augment.generators)

T

tensor_to_list() (in module kitanaqa.trainer.train) tokenizer_name_or_path (kitanaqa.trainer.arguments.ModelArguments attribute) train_file (kitanaqa.trainer.alum_squad_processor.AlumSquadProcessor attribute) (kitanaqa.trainer.alum_squad_processor.AlumSquadV1Processor attribute) (kitanaqa.trainer.alum_squad_processor.AlumSquadV2Processor attribute)

train_file_path (kitanaqa.trainer.arguments.ModelArguments attribute) Trainer (class in kitanaqa.trainer.train) training_step() (kitanaqa.trainer.train.Trainer method), [1]

V

validate_inputs() (in module kitanaqa.augment.term_replacement)

verbose_logging (kitanaqa.trainer.arguments.ModelArguments attribute) version_2_with_negative (kitanaqa.trainer.arguments.ModelArguments attribute)