Ubuntu
commited on
Commit
•
aec05c4
1
Parent(s):
672b0f2
added model
Browse files- 0_Transformer/config.json +30 -0
- 0_Transformer/pytorch_model.bin +3 -0
- 0_Transformer/sentence_bert_config.json +4 -0
- 0_Transformer/special_tokens_map.json +1 -0
- 0_Transformer/tokenizer_config.json +1 -0
- 0_Transformer/vocab.txt +0 -0
- 1_Pooling/config.json +7 -0
- config.json +3 -0
- modules.json +14 -0
- similarity_evaluation_sts-dev_results.csv +9 -0
- similarity_evaluation_sts-test_results.csv +2 -0
0_Transformer/config.json
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "output-no-git/mBERT/pt-e5-select/0_Transformer",
|
3 |
+
"architectures": [
|
4 |
+
"BertModel"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.1,
|
7 |
+
"directionality": "bidi",
|
8 |
+
"gradient_checkpointing": false,
|
9 |
+
"hidden_act": "gelu",
|
10 |
+
"hidden_dropout_prob": 0.1,
|
11 |
+
"hidden_size": 768,
|
12 |
+
"initializer_range": 0.02,
|
13 |
+
"intermediate_size": 3072,
|
14 |
+
"layer_norm_eps": 1e-12,
|
15 |
+
"max_position_embeddings": 512,
|
16 |
+
"model_type": "bert",
|
17 |
+
"num_attention_heads": 12,
|
18 |
+
"num_hidden_layers": 12,
|
19 |
+
"pad_token_id": 0,
|
20 |
+
"pooler_fc_size": 768,
|
21 |
+
"pooler_num_attention_heads": 12,
|
22 |
+
"pooler_num_fc_layers": 3,
|
23 |
+
"pooler_size_per_head": 128,
|
24 |
+
"pooler_type": "first_token_transform",
|
25 |
+
"position_embedding_type": "absolute",
|
26 |
+
"transformers_version": "4.6.0.dev0",
|
27 |
+
"type_vocab_size": 2,
|
28 |
+
"use_cache": true,
|
29 |
+
"vocab_size": 119547
|
30 |
+
}
|
0_Transformer/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23bfae162345fb7e21e6c27040dae85f32bd056dc51024c51995ce19540c87b8
|
3 |
+
size 711500343
|
0_Transformer/sentence_bert_config.json
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"max_seq_length": null,
|
3 |
+
"do_lower_case": false
|
4 |
+
}
|
0_Transformer/special_tokens_map.json
ADDED
@@ -0,0 +1 @@
|
|
|
1 |
+
{"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
|
0_Transformer/tokenizer_config.json
ADDED
@@ -0,0 +1 @@
|
|
|
1 |
+
{"do_lower_case": false, "do_basic_tokenize": true, "never_split": null, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "tokenize_chinese_chars": true, "strip_accents": null, "truncation": true, "model_max_length": 512, "special_tokens_map_file": null, "tokenizer_file": "/home/ubuntu/.cache/huggingface/transformers/46880f3b0081fda494a4e15b05787692aa4c1e21e0ff2428ba8b14d4eda0784d.b33e51591f94f17c238ee9b1fac75b96ff2678cbaed6e108feadb3449d18dc24", "name_or_path": "output-no-git/mBERT/pt-e5-select/0_Transformer", "max_length": 510}
|
0_Transformer/vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
1_Pooling/config.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"word_embedding_dimension": 768,
|
3 |
+
"pooling_mode_cls_token": false,
|
4 |
+
"pooling_mode_mean_tokens": true,
|
5 |
+
"pooling_mode_max_tokens": false,
|
6 |
+
"pooling_mode_mean_sqrt_len_tokens": false
|
7 |
+
}
|
config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"__version__": "1.1.0"
|
3 |
+
}
|
modules.json
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[
|
2 |
+
{
|
3 |
+
"idx": 0,
|
4 |
+
"name": "0",
|
5 |
+
"path": "0_Transformer",
|
6 |
+
"type": "sentence_transformers.models.Transformer"
|
7 |
+
},
|
8 |
+
{
|
9 |
+
"idx": 1,
|
10 |
+
"name": "1",
|
11 |
+
"path": "1_Pooling",
|
12 |
+
"type": "sentence_transformers.models.Pooling"
|
13 |
+
}
|
14 |
+
]
|
similarity_evaluation_sts-dev_results.csv
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
|
2 |
+
0,1000,0.8066819860834835,0.7717123426064139,0.8071555390390152,0.7805133399101568,0.8071183612045486,0.7795454576095555,0.6702574668713819,0.563015710809403
|
3 |
+
0,-1,0.8623863322491688,0.8366184677468235,0.8478859659193637,0.840685074857325,0.8482289129922826,0.8408182354953139,0.7393878246731052,0.6594683325907266
|
4 |
+
1,1000,0.8926176031326222,0.8514790317201159,0.8620403836695407,0.8509810842300843,0.8614773079998486,0.8509621509177535,0.7697762418589701,0.6897989607168812
|
5 |
+
1,-1,0.9156024325414877,0.8916452397633556,0.8961670388206086,0.8885776951208231,0.8956834113063812,0.8882068376301452,0.7899962410250485,0.6877511727328209
|
6 |
+
2,1000,0.9321394074160259,0.9053347957307508,0.907596601927006,0.903358857378851,0.906851229223818,0.902738739909935,0.8325162776154821,0.740531983781187
|
7 |
+
2,-1,0.9368569426813244,0.9161208826011831,0.9121968309568799,0.911992906822889,0.9113977933544264,0.9111703874832967,0.8417083571523954,0.7595743890365477
|
8 |
+
3,1000,0.9474683986619091,0.9257585665590768,0.9201330033560583,0.9212963908665958,0.919318843872525,0.9209813020404272,0.8652464181835349,0.7897202202524914
|
9 |
+
3,-1,0.948476915224546,0.9255690872348762,0.9183924126108384,0.9200899430186467,0.917452736627362,0.9192259151487764,0.8666736859618032,0.7924544449795947
|
similarity_evaluation_sts-test_results.csv
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
1 |
+
epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
|
2 |
+
-1,-1,0.8096149879355693,0.7879266446025123,0.7898945538229822,0.7795216042250593,0.7897610221737497,0.7793498345544976,0.7234097326687203,0.6556548888964406
|