MCPcopy Create free account

hub / github.com/castorini/castor / functions

Functions334 in github.com/castorini/castor

↓ 24 callersMethodget_evaluator
(dataset_cls, model, embedding, data_loader, batch_size, device, nce=False, keep_results=False)
common/evaluation.py:38
↓ 13 callersMethodevaluate
(self, evaluator, dataset_name)
common/trainers/trainer.py:30
↓ 11 callersMethoditers
:param path: directory containing train, test, dev files :param vectors_name: name of word vectors file :param vectors_cache:
datasets/sst.py:39
↓ 10 callersFunctionstopped
remove stop words from given sentences (questions|answers)
sm_cnn/external_features.py:21
↓ 9 callersFunctionread_in_data
(datapath, set_name, file, stop_and_stem=False, stop_punct=False, dash_split=False)
idf_baseline/qa-data-only-idf.py:17
↓ 8 callersMethodortho_weight
Random orthogonal weights Used by norm_weights(below), in which case, we are ensuring that the rows are orthogonal (i
esim/model.py:152
↓ 8 callersMethodtrain
(self, epochs)
common/trainers/trainer.py:46
↓ 7 callersMethodget_scores
Get the scores used to evaluate the model. Should return ([score1, score2, ..], [score1_name, score2_name, ...]). The first s
common/evaluators/evaluator.py:20
↓ 7 callersMethodget_sentence_embeddings
(self, batch)
common/trainers/trainer.py:38
↓ 7 callersMethodget_sentence_embeddings
(self, batch)
common/evaluators/evaluator.py:15
↓ 7 callersFunctionsave_checkpoint
(epoch, arch, state_dict, optimizer_state, eval_metric, filename)
utils/serialization.py:7
↓ 6 callersMethod_unsqueeze
(self, tensor)
sm_cnn/model.py:42
↓ 6 callersFunctioncompute_idf_weighted_overlap
returns idf weighted overlap
sm_cnn/external_features.py:113
↓ 6 callersFunctioncompute_overlap
returns simple overlap between document pairs
sm_cnn/external_features.py:100
↓ 6 callersMethodget_dataset
(dataset_name, word_vectors_dir, word_vectors_file, batch_size, device, castor_dir="./", utils_trecqa="utils/t
common/dataset.py:36
↓ 6 callersFunctionget_map_mrr
Get the map and mrr using the trec_eval utility. qids, predictions, labels should have the same length. device is not a required paramete
utils/relevancy_metrics.py:6
↓ 6 callersMethodget_trainer
(dataset_name, model, embedding, train_loader, trainer_config, train_evaluator, test_evaluator, dev_evaluator=
common/train.py:38
↓ 6 callersMethodsplits
(cls, question_id, question_field, answer_field, external_field, label_field, root='.data', tra
sm_cnn/trec_dataset.py:7
↓ 5 callersFunctionload_checkpoint
(filename)
utils/serialization.py:21
↓ 5 callersFunctionmergeBuffers
(recBuffers, recLength)
anserini_dependency/js/recorderWorker.js:89
↓ 5 callersMethodparse
(self, sentence)
sm_cnn/bridge.py:64
↓ 5 callersMethodsplits
(cls, path, train=os.path.join('SST', 'stsa.fine.phrases.train'), validation=os.path.join('SST'
datasets/sst.py:31
↓ 4 callersMethodadd_setting
(self, setting)
idf_baseline/experimental_settings.py:36
↓ 4 callersFunctionget_batch
(question, answer, ext_feat, size)
utils/nce_neighbors.py:45
↓ 4 callersMethodtrain
(self, epochs)
nce/nce_pairwise_mp/trainers/qa_trainer.py:203
↓ 4 callersFunctionwriteString
(view, offset, string)
anserini_dependency/js/recorderWorker.js:121
↓ 3 callersMethod_run_cmd
(self, cmd)
idf_baseline/experimental_settings.py:50
↓ 3 callersMethodattention_softmax3d
(self, raw_attentions)
decatt/model.py:72
↓ 3 callersFunctionget_args
()
sm_cnn/args.py:3
↓ 3 callersFunctionortho_weight
Random orthogonal weights Used by norm_weights(below), in which case, we are ensuring that the rows are orthogonal (i.e W = U \Sigma
esim/model.py:12
↓ 3 callersMethodset_vectors
(cls, field, vector_path)
datasets/castor_dataset.py:59
↓ 3 callersFunctionstem_non_stop_words
(sentence)
sm_cnn/external_features.py:186
↓ 3 callersFunctionwrite_out_idf_sum_similarities
(qids, questions, answers, term_idfs, outfile, dataset)
idf_baseline/qa-data-only-idf.py:109
↓ 2 callersFunctionPhi
(x)
vdpwi/utils/preprocess.py:22
↓ 2 callersMethod__init__
(self, config)
vdpwi/model.py:19
↓ 2 callersMethod__init__
(self, device, in_dim, mem_dim)
esim/model.py:75
↓ 2 callersMethod__init__
(self, word_model, **config)
conv_rnn/model.py:14
↓ 2 callersMethod_get_blocks_for_sentence
(self, sent)
mp_cnn/lite_model.py:43
↓ 2 callersMethod_get_blocks_for_sentence
(self, sent)
mp_cnn/model.py:93
↓ 2 callersMethod_transformation_input
(self, embed_sent)
decatt/model.py:77
↓ 2 callersFunctionadd_vocab
(tok_filename, vocab)
vdpwi/utils/preprocess.py:61
↓ 2 callersMethodcompare
Apply a feed forward network to compare o ne sentence to its soft alignment with the other. :param sentence: embedded and
decatt/model.py:129
↓ 2 callersFunctioncompute_overlap_features
(questions, answers, word2df=None, stoplist=None)
sm_cnn/overlap_features.py:36
↓ 2 callersFunctioncompute_overlap_features
(questions, answers, word2df=None, stoplist=None)
nce/nce_pairwise_sm/overlap_features.py:36
↓ 2 callersMethodconcat_attention
(self, sent1, sent2, word_to_doc_count=None, raw_sent1=None, raw_sent2=None)
mp_cnn/model.py:176
↓ 2 callersMethodcreate_mask
(self, sent)
esim/model.py:237
↓ 2 callersFunctionencodeWAV
(samples, mono)
anserini_dependency/js/recorderWorker.js:127
↓ 2 callersFunctionevaluate
(loader, dev=True)
conv_rnn/train.py:90
↓ 2 callersFunctionevaluate_dataset
(split_name, dataset_cls, model, embedding, loader, batch_size, device)
vdpwi/__main__.py:18
↓ 2 callersFunctionevaluate_dataset
(split_name, dataset_cls, model, embedding, loader, batch_size, device, keep_results=False)
esim/__main__.py:31
↓ 2 callersFunctionevaluate_dataset
(split_name, dataset_cls, model, embedding, loader, batch_size, device, keep_results=False)
decatt/__main__.py:31
↓ 2 callersFunctionevaluate_dataset
(split_name, dataset_cls, model, embedding, loader, batch_size, device, keep_results=False)
mp_cnn/__main__.py:32
↓ 2 callersFunctionevaluate_dataset
(split_name, dataset_cls, model, embedding, loader, batch_size, device, keep_results=False)
sse/__main__.py:31
↓ 2 callersFunctionfetch_indices
(name)
vdpwi/data.py:62
↓ 2 callersMethodgetRankedPassages
Calls RetrieveSentences.getRankedPassages Parameters ---------- query : str The query to be searched in
anserini_dependency/RetrieveSentences.py:47
↓ 2 callersMethodgetTermIDF
(String term)
idf_baseline/src/main/java/ai/castor/idf/FetchTermIDF.java:66
↓ 2 callersMethodgetTermIdfJSON
Calls RetrieveSentences.getTermIdfJSON
anserini_dependency/RetrieveSentences.py:70
↓ 2 callersFunctionget_args
()
nce/nce_pairwise_sm/args.py:3
↓ 2 callersFunctionget_nearest_neg_id
(pos_feature, neg_dict, distance="cosine", k=1, weight=False)
utils/nce_neighbors.py:10
↓ 2 callersFunctionget_qadata_only_idf
returns idf weights computed over all question answer pairs in the dataset
sm_cnn/external_features.py:40
↓ 2 callersFunctionget_random_neg_id
(q2neg, qid_i, k=8)
utils/nce_neighbors.py:36
↓ 2 callersFunctionget_source_corpus_idf
fetches idf weights from source corpus (disks1-5+aquaint|wikipedia) index, for all the qa pairs
sm_cnn/external_features.py:55
↓ 2 callersFunctionload_data
(dname)
sm_cnn/overlap_features.py:9
↓ 2 callersFunctionload_data
(dname)
nce/nce_pairwise_sm/overlap_features.py:9
↓ 2 callersMethodload_sst_sets
(cls, dirname, fmt="stsa.fine.{}")
conv_rnn/data.py:50
↓ 2 callersFunctionpad_side
(idx)
vdpwi/model.py:8
↓ 2 callersFunctionphi
(zeta)
vdpwi/utils/preprocess.py:20
↓ 2 callersFunctionpredict
(dataset, test_mode, dataset_iter)
sm_cnn/main.py:79
↓ 2 callersFunctionpredict
(test_mode, dataset_iter)
nce/nce_pairwise_sm/main.py:62
↓ 2 callersFunctionread_labels
(filename)
vdpwi/data.py:72
↓ 2 callersMethodrerank_candidate_answers
(self, question, answers, idf_json)
sm_cnn/bridge.py:69
↓ 2 callersMethodrun
runs a particular combination of settings
idf_baseline/experimental_settings.py:85
↓ 2 callersFunctionset_vectors
(field, vector_path)
sm_cnn/train.py:22
↓ 2 callersMethodsick_config
()
vdpwi/data.py:36
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/snli.py:41
↓ 2 callersMethodsplits
(cls, path, train='train', test='test', **kwargs)
datasets/msrvid.py:48
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/pit2015.py:33
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/wikiqa.py:33
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/sick.py:48
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/quora.py:41
↓ 2 callersMethodsplits
(cls, path, train='train-all', validation='raw-dev', test='raw-test', **kwargs)
datasets/trecqa.py:33
↓ 2 callersMethodsplits
(cls, path, train='train', validation='dev', test='test', **kwargs)
datasets/sts2014.py:49
↓ 2 callersFunctionstemmed
reduce sentence terms to stemmed representations
sm_cnn/external_features.py:31
↓ 2 callersFunctiontrain
(**kwargs)
conv_rnn/train.py:55
↓ 1 callersFunctionURL_maxF1_eval
(predict_result, test_data_label)
common/evaluators/pit2015_evaluator.py:6
↓ 1 callersMethod__init__
(self, config)
nce/nce_pairwise_sm/model.py:30
↓ 1 callersMethod_add_layers
(self)
mp_cnn/model.py:35
↓ 1 callersMethod_algo_1_horiz_comp
(self, sent1_block_a, sent2_block_a)
mp_cnn/lite_model.py:60
↓ 1 callersMethod_algo_1_horiz_comp
(self, sent1_block_a, sent2_block_a)
mp_cnn/model.py:123
↓ 1 callersMethod_algo_2_vert_comp
(self, sent1_block_a, sent2_block_a)
mp_cnn/lite_model.py:86
↓ 1 callersMethod_algo_2_vert_comp
(self, sent1_block_a, sent2_block_a, sent1_block_b, sent2_block_b)
mp_cnn/model.py:150
↓ 1 callersMethod_get_n_feats
(self)
mp_cnn/model.py:79
↓ 1 callersMethod_run_eval
(self)
idf_baseline/experimental_settings.py:57
↓ 1 callersMethod_setup_combinations
(self)
idf_baseline/experimental_settings.py:41
↓ 1 callersMethod_update
(self, carry_idx)
conv_rnn/train.py:30
↓ 1 callersMethodaggregate
Aggregate the representations induced from both sentences and their representations :param v1: tensor with shape (batch, time
esim/model.py:201
↓ 1 callersMethodaggregate
Aggregate the representations induced from both sentences and their representations :param v1: tensor with shape (batch, tim
decatt/model.py:144
↓ 1 callersMethodattend
Compute inter-sentence attention. This is step 1 (attend) in the paper :param sent1: tensor in shape (batch, time_steps, num_units),
decatt/model.py:100
↓ 1 callersMethodbase_config
()
vdpwi/data.py:11
↓ 1 callersFunctionbuild_vector_cache
(glove_filename, vec_cache_filename, vocab)
vdpwi/utils/preprocess.py:10
next →1–100 of 334, ranked by callers