Index

A

ALBERTEncoder (class in src.encoders)

B

BasicTokenizer (class in src.tokenizers.bert_tokenization)

BERTEncoder (class in src.encoders) build() (src.prebuilt_index.SimpleNNIndex method), [1]

C

convert_by_vocab() (in module src.tokenizers.bert_tokenization) convert_ids_to_tokens() (in module src.tokenizers.bert_tokenization) (src.tokenizers.bert_tokenization.FullTokenizer method)

convert_to_unicode() (in module src.tokenizers.bert_tokenization) convert_tokens_to_ids() (in module src.tokenizers.bert_tokenization) (src.tokenizers.bert_tokenization.FullTokenizer method) create_df() (src.data_handler.kb_handler.kb method)

D

download_emb_index() (src.minio_handler.MinioClient method)

download_model_weights() (src.minio_handler.MinioClient method)

E

encode() (src.encoders.ALBERTEncoder method) (src.encoders.BERTEncoder method) (src.encoders.Encoder method), [1] (src.encoders.USEEncoder method), [1]

Encoder (class in src.encoders), [1] eval_model() (in module src.finetune.eval) export_encoder() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1]

F

finetune() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1] finetune_weights() (src.encoders.ALBERTEncoder method) (src.encoders.BERTEncoder method) (src.encoders.Encoder method), [1] (src.encoders.USEEncoder method), [1]

FullTokenizer (class in src.tokenizers.bert_tokenization)

G

gen() (in module src.finetune.generators) generate_mappings() (in module src.data_handler.kb_handler) get_eval_dict() (in module src.finetune.eval)

get_ids() (in module src.tokenizers.bert_tokenization) get_masks() (in module src.tokenizers.bert_tokenization) get_segments() (in module src.tokenizers.bert_tokenization) GoldenRetriever (class in src.models), [1]

H

hard_triplet_generator() (in module src.finetune.generators)

I

init_signatures() (src.encoders.ALBERTEncoder method) (src.encoders.BERTEncoder method) (src.encoders.USEEncoder method), [1]

J

json() (src.data_handler.kb_handler.kb method)

K

kb (class in src.data_handler.kb_handler)

kb_handler (class in src.data_handler.kb_handler)

L

load() (src.prebuilt_index.SimpleNNIndex class method), [1] load_es_kb() (src.data_handler.kb_handler.kb_handler method)

load_kb() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1] load_vocab() (in module src.tokenizers.bert_tokenization)

M

make_bucket() (src.minio_handler.MinioClient method) make_query() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1]

MinioClient (class in src.minio_handler) Model (class in src.models), [1] mrr() (in module src.finetune.eval)

P

parse_csv() (src.data_handler.kb_handler.kb_handler method) parse_df() (src.data_handler.kb_handler.kb_handler method) parse_pdf() (src.data_handler.kb_handler.kb_handler method) parse_text() (src.data_handler.kb_handler.kb_handler method) predict() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1]

preprocess_one_str() (in module src.tokenizers.bert_tokenization) preprocess_str() (in module src.tokenizers.bert_tokenization) preview() (src.data_handler.kb_handler.kb_handler method) printable_text() (in module src.tokenizers.bert_tokenization)

Q

query() (src.prebuilt_index.SimpleNNIndex method), [1]

R

random_triplet_generator() (in module src.finetune.generators) recall_at_n() (in module src.finetune.eval) restore_encoder() (src.models.GoldenRetriever method), [1] (src.models.Model method), [1]

restore_weights() (src.encoders.ALBERTEncoder method) (src.encoders.BERTEncoder method) (src.encoders.Encoder method), [1] (src.encoders.USEEncoder method), [1] rm_bucket() (src.minio_handler.MinioClient method)

S

save() (src.prebuilt_index.SimpleNNIndex method), [1] save_weights() (src.encoders.ALBERTEncoder method) (src.encoders.BERTEncoder method) (src.encoders.Encoder method), [1] (src.encoders.USEEncoder method), [1] SimpleNNIndex (class in src.prebuilt_index), [1] src (module) src.data_handler.kb_handler (module) src.elasticsearch.create_doc_index (module), [1]

src.encoders (module), [1] src.finetune.eval (module) src.finetune.generators (module) src.loss_functions (module), [1] src.minio_handler (module) src.models (module), [1] src.prebuilt_index (module), [1] src.tokenizers (module) src.tokenizers.bert_tokenization (module)

T

tokenize() (src.tokenizers.bert_tokenization.BasicTokenizer method) (src.tokenizers.bert_tokenization.FullTokenizer method) (src.tokenizers.bert_tokenization.WordpieceTokenizer method)

triplet_loss() (in module src.loss_functions), [1] truncate_str() (in module src.tokenizers.bert_tokenization)

U

unique_indexing() (in module src.data_handler.kb_handler) upload_docs() (in module src.elasticsearch.create_doc_index), [1]

upload_emb_index() (src.minio_handler.MinioClient method) upload_model_weights() (src.minio_handler.MinioClient method) USEEncoder (class in src.encoders), [1]

V

validate_case_matches_checkpoint() (in module src.tokenizers.bert_tokenization)

W

whitespace_tokenize() (in module src.tokenizers.bert_tokenization)

WordpieceTokenizer (class in src.tokenizers.bert_tokenization)