Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenMatch/UniVL-DR
/ functions
Functions
115 in github.com/OpenMatch/UniVL-DR
⨍
Functions
115
◇
Types & classes
9
↓ 6 callers
Function
load_file
(path, txt=True, img=True)
UniVL-DR/data.py:217
↓ 6 callers
Function
load_file
(path, txt=True, img=True)
CLIP-DPR/data.py:146
↓ 3 callers
Function
compute_metrics
Compute MRR metric Args: p_qids_to_relevant_passageids (dict): dictionary of query-passage mapping Dict as read in with load_reference
UniVL-DR/msmarco_eval.py:114
↓ 3 callers
Function
compute_metrics
Compute MRR metric Args: p_qids_to_relevant_passageids (dict): dictionary of query-passage mapping Dict as read in with load_reference
CLIP-DPR/msmarco_eval.py:114
↓ 3 callers
Method
encode_img
(self, idx)
UniVL-DR/data.py:48
↓ 3 callers
Function
gen_embeddings
(model, valid_reader, outpath)
UniVL-DR/gen_embeddings.py:150
↓ 3 callers
Function
gen_embeddings
(model, valid_reader, outpath)
CLIP-DPR/gen_embeddings.py:150
↓ 3 callers
Function
load_docs
(path)
UniVL-DR/data.py:242
↓ 3 callers
Function
load_docs
(path)
CLIP-DPR/data.py:166
↓ 2 callers
Function
convert_to_string_id
(result_dict)
UniVL-DR/retrieval.py:20
↓ 2 callers
Function
convert_to_string_id
(result_dict)
CLIP-DPR/retrieval.py:20
↓ 2 callers
Method
encode_img
(self, idx)
CLIP-DPR/data.py:45
↓ 2 callers
Function
get_qrels
(example)
data/process_data.py:21
↓ 2 callers
Function
load_caps
(path)
UniVL-DR/data.py:251
↓ 2 callers
Function
load_caps
(path)
CLIP-DPR/data.py:175
↓ 2 callers
Function
process_example
(example)
data/process_data.py:7
↓ 1 callers
Function
EvalDevQuery
(query_positive_id, ctx_idxs)
UniVL-DR/retrieval.py:34
↓ 1 callers
Function
EvalDevQuery
(query_positive_id, ctx_idxs)
CLIP-DPR/retrieval.py:34
↓ 1 callers
Method
_ensure_tsv_opened
(self)
UniVL-DR/visual.py:45
↓ 1 callers
Method
_ensure_tsv_opened
(self)
CLIP-DPR/visual.py:45
↓ 1 callers
Function
_warmup_lr
(base_lr, warmup_length, step)
UniVL-DR/gen_embeddings.py:39
↓ 1 callers
Function
_warmup_lr
(base_lr, warmup_length, step)
UniVL-DR/train.py:39
↓ 1 callers
Function
_warmup_lr
(base_lr, warmup_length, step)
CLIP-DPR/gen_embeddings.py:39
↓ 1 callers
Function
_warmup_lr
(base_lr, warmup_length, step)
CLIP-DPR/train.py:39
↓ 1 callers
Function
assign_learning_rate
(optimizer, new_lr)
UniVL-DR/gen_embeddings.py:34
↓ 1 callers
Function
assign_learning_rate
(optimizer, new_lr)
UniVL-DR/train.py:34
↓ 1 callers
Function
assign_learning_rate
(optimizer, new_lr)
CLIP-DPR/gen_embeddings.py:34
↓ 1 callers
Function
assign_learning_rate
(optimizer, new_lr)
CLIP-DPR/train.py:34
↓ 1 callers
Function
compute_metrics_from_files
Compute MRR metric Args: p_path_to_reference_file (str): path to reference file. Reference file should contain lines in the following
UniVL-DR/msmarco_eval.py:148
↓ 1 callers
Function
compute_metrics_from_files
Compute MRR metric Args: p_path_to_reference_file (str): path to reference file. Reference file should contain lines in the following
CLIP-DPR/msmarco_eval.py:147
↓ 1 callers
Function
convert_models_to_fp32
(model)
UniVL-DR/train.py:29
↓ 1 callers
Function
convert_models_to_fp32
(model)
CLIP-DPR/train.py:29
↓ 1 callers
Function
cosine_lr
(optimizer, base_lr, warmup_length, steps)
UniVL-DR/train.py:42
↓ 1 callers
Function
cosine_lr
(optimizer, base_lr, warmup_length, steps)
CLIP-DPR/train.py:42
↓ 1 callers
Function
encode_img
(idx)
UniVL-DR/get_imgs.py:29
↓ 1 callers
Method
encode_img
(self, idx)
UniVL-DR/gen_embeddings.py:75
↓ 1 callers
Method
encode_img
(self, idx)
CLIP-DPR/gen_embeddings.py:75
↓ 1 callers
Function
eval_loss
(model, valid_reader)
UniVL-DR/train.py:61
↓ 1 callers
Function
eval_loss
(model, loss_function, valid_reader)
CLIP-DPR/train.py:61
↓ 1 callers
Function
load_candidate
Load candidate data from a file. Args:path_to_candidate (str): path to file to load. Returns:qid_to_ranked_candidate_passages (dict): dictiona
UniVL-DR/msmarco_eval.py:72
↓ 1 callers
Function
load_candidate
Load candidate data from a file. Args:path_to_candidate (str): path to file to load. Returns:qid_to_ranked_candidate_passages (dict): dictiona
CLIP-DPR/msmarco_eval.py:72
↓ 1 callers
Function
load_candidate_from_stream
Load candidate data from a stream. Args:f (stream): stream to load. Returns:qid_to_ranked_candidate_passages (dict): dictionary mapping from q
UniVL-DR/msmarco_eval.py:48
↓ 1 callers
Function
load_candidate_from_stream
Load candidate data from a stream. Args:f (stream): stream to load. Returns:qid_to_ranked_candidate_passages (dict): dictionary mapping from q
CLIP-DPR/msmarco_eval.py:48
↓ 1 callers
Function
load_file
(path, txt=True, img=True)
UniVL-DR/retrieval.py:118
↓ 1 callers
Function
load_file
(path, txt=True, img=True)
CLIP-DPR/retrieval.py:118
↓ 1 callers
Function
load_file
(path)
CLIP-DPR/get_hard_negs.py:20
↓ 1 callers
Function
load_file
(path)
CLIP-DPR/get_retrieval_results.py:20
↓ 1 callers
Function
load_file
(path)
CLIP-DPR/get_hard_negs_all.py:20
↓ 1 callers
Function
load_queries
(path)
UniVL-DR/gen_embeddings.py:184
↓ 1 callers
Function
load_queries
(path)
CLIP-DPR/gen_embeddings.py:184
↓ 1 callers
Function
load_reference
Load Reference reference relevant passages Args:path_to_reference (str): path to a file to load. Returns:qids_to_relevant_passageids (dict): d
UniVL-DR/msmarco_eval.py:38
↓ 1 callers
Function
load_reference
Load Reference reference relevant passages Args:path_to_reference (str): path to a file to load. Returns:qids_to_relevant_passageids (dict): d
CLIP-DPR/msmarco_eval.py:38
↓ 1 callers
Function
load_reference_from_stream
Load Reference reference relevant passages Args:f (stream): stream to load. Returns:qids_to_relevant_passageids (dict): dictionary mapping fro
UniVL-DR/msmarco_eval.py:18
↓ 1 callers
Function
load_reference_from_stream
Load Reference reference relevant passages Args:f (stream): stream to load. Returns:qids_to_relevant_passageids (dict): dictionary mapping fro
CLIP-DPR/msmarco_eval.py:18
↓ 1 callers
Function
main
Command line: python msmarco_eval_ranking.py <path_to_reference_file> <path_to_candidate_file>
UniVL-DR/msmarco_eval.py:174
↓ 1 callers
Function
main
Command line: python msmarco_eval_ranking.py <path_to_reference_file> <path_to_candidate_file>
CLIP-DPR/msmarco_eval.py:173
↓ 1 callers
Method
num_rows
(self)
UniVL-DR/visual.py:30
↓ 1 callers
Method
num_rows
(self)
CLIP-DPR/visual.py:30
↓ 1 callers
Function
quality_checks_qids
Perform quality checks on the dictionaries Args: p_qids_to_relevant_passageids (dict): dictionary of query-passage mapping Dict as re
UniVL-DR/msmarco_eval.py:83
↓ 1 callers
Function
quality_checks_qids
Perform quality checks on the dictionaries Args: p_qids_to_relevant_passageids (dict): dictionary of query-passage mapping Dict as re
CLIP-DPR/msmarco_eval.py:83
↓ 1 callers
Method
seek
(self, pos)
UniVL-DR/visual.py:33
↓ 1 callers
Method
seek
(self, pos)
CLIP-DPR/visual.py:33
↓ 1 callers
Function
set_seed
(args)
UniVL-DR/train.py:55
↓ 1 callers
Function
set_seed
(args)
CLIP-DPR/train.py:55
↓ 1 callers
Function
train
(train_reader, valid_reader, model)
UniVL-DR/train.py:105
↓ 1 callers
Function
train
(train_reader, valid_reader, model)
CLIP-DPR/train.py:102
Method
Collector
(self, batch)
UniVL-DR/gen_embeddings.py:85
Method
Collector
(self, batch)
UniVL-DR/gen_embeddings.py:125
Method
Collector
(self, batch)
UniVL-DR/data.py:58
Method
Collector
(self, batch)
CLIP-DPR/gen_embeddings.py:85
Method
Collector
(self, batch)
CLIP-DPR/gen_embeddings.py:125
Method
Collector
(self, batch)
CLIP-DPR/data.py:55
Method
__del__
(self)
UniVL-DR/visual.py:20
Method
__del__
(self)
CLIP-DPR/visual.py:20
Method
__getitem__
(self, index)
UniVL-DR/gen_embeddings.py:102
Method
__getitem__
(self, index)
UniVL-DR/gen_embeddings.py:138
Method
__getitem__
(self, pos)
UniVL-DR/visual.py:38
Method
__getitem__
(self, index)
UniVL-DR/data.py:136
Method
__getitem__
(self, index)
CLIP-DPR/gen_embeddings.py:102
Method
__getitem__
(self, index)
CLIP-DPR/gen_embeddings.py:138
Method
__getitem__
(self, pos)
CLIP-DPR/visual.py:38
Method
__getitem__
(self, index)
CLIP-DPR/data.py:100
Method
__init__
(self, args, preprocess, captions=None)
UniVL-DR/gen_embeddings.py:55
Method
__init__
(self, data, max_len)
UniVL-DR/gen_embeddings.py:115
Method
__init__
(self, tsv_file, total_num)
UniVL-DR/visual.py:14
Method
__init__
(self, args, preprocess_fn, data, docs, captions, shuffle)
UniVL-DR/data.py:18
Method
__init__
(self, args, preprocess, captions=None)
CLIP-DPR/gen_embeddings.py:55
Method
__init__
(self, data, max_len)
CLIP-DPR/gen_embeddings.py:115
Method
__init__
(self, encoder)
CLIP-DPR/models.py:11
Method
__init__
(self, tsv_file, total_num)
CLIP-DPR/visual.py:14
Method
__init__
(self, args, preprocess_fn, data, docs, captions, shuffle)
CLIP-DPR/data.py:18
Method
__len__
(self)
UniVL-DR/gen_embeddings.py:72
Method
__len__
(self)
UniVL-DR/gen_embeddings.py:119
Method
__len__
(self)
UniVL-DR/visual.py:41
Method
__len__
(self)
UniVL-DR/data.py:44
Method
__len__
(self)
CLIP-DPR/gen_embeddings.py:72
Method
__len__
(self)
CLIP-DPR/gen_embeddings.py:119
Method
__len__
(self)
CLIP-DPR/visual.py:41
Method
__len__
(self)
CLIP-DPR/data.py:41
Method
__repr__
(self)
UniVL-DR/visual.py:27
next →
1–100 of 115, ranked by callers