forked from nguyenvo09/EMNLP2020
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setting_keywords.py
48 lines (35 loc) · 1.36 KB
/
setting_keywords.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
class KeyWordSettings(object):
Doc_cID = "doc_cid"
Doc_URL = "doc_ciurl"
Doc_cLabel = "doc_clabel"
Doc_wImages = "doc_wimages"
Doc_wContent = "doc_wcontent"
Relevant_Score = "relevant_score"
Query_id = "qid"
Query_TweetID = "qtweetid"
Query_Images = "query_images"
Ranked_Docs = "ranked_docs"
Query_Content = "query_content"
Query_lens = "query_lens"
Doc_lens = "docs_lens"
Query_Idf = "query_idf"
QueryImagesIndices = "query_images_indices"
DocImagesIndices = "doc_images_indices"
NegDocImagesIndices = "neg_doc_images_indices"
QueryIDs = "query_ids"
DocIDs = "doc_ids"
UseVisual = "use_visual"
FullLeftImgTensor = "full_left_images_tensor"
FullRightImgTensor = "full_right_images_tensor"
ImageLoaderKey = "image_loader"
OutputRankingKey = "output_ranking"
Index2Query = "index2queries"
Index2Doc = "index2docs"
Test2Hard = "test2_hard"
Test3Hard = "test3_hard"
QueryCountVal = [1116, 1000, 187, 1159]
QueryCountTest = [1001, 1164, 1118, 187, 156, 1160, 1500]
UseCuda = "use_cuda"
LOSS_FUNCTIONS = ('pointwise', 'bpr', 'hinge', 'adaptive_hinge', "single_pointwise_square_loss", "pce", "bce",
"cosine_max_margin_loss_dvsh", "cross_entropy", "vanilla_cross_entropy", "regression_loss",
"masked_cross_entropy")