jfarray commited on
Commit
684d3fa
1 Parent(s): 0cb8611

Add new SentenceTransformer model.

Browse files
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
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
+ }
README.md ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: sentence-similarity
3
+ tags:
4
+ - sentence-transformers
5
+ - feature-extraction
6
+ - sentence-similarity
7
+ ---
8
+
9
+ # {MODEL_NAME}
10
+
11
+ This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
12
+
13
+ <!--- Describe your model here -->
14
+
15
+ ## Usage (Sentence-Transformers)
16
+
17
+ Using this model becomes easy when you have [sentence-transformers](https://www.SBERT.net) installed:
18
+
19
+ ```
20
+ pip install -U sentence-transformers
21
+ ```
22
+
23
+ Then you can use the model like this:
24
+
25
+ ```python
26
+ from sentence_transformers import SentenceTransformer
27
+ sentences = ["This is an example sentence", "Each sentence is converted"]
28
+
29
+ model = SentenceTransformer('{MODEL_NAME}')
30
+ embeddings = model.encode(sentences)
31
+ print(embeddings)
32
+ ```
33
+
34
+
35
+
36
+ ## Evaluation Results
37
+
38
+ <!--- Describe how your model was evaluated -->
39
+
40
+ For an automated evaluation of this model, see the *Sentence Embeddings Benchmark*: [https://seb.sbert.net](https://seb.sbert.net?model_name={MODEL_NAME})
41
+
42
+
43
+ ## Training
44
+ The model was trained with the parameters:
45
+
46
+ **DataLoader**:
47
+
48
+ `torch.utils.data.dataloader.DataLoader` of length 11 with parameters:
49
+ ```
50
+ {'batch_size': 15, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
51
+ ```
52
+
53
+ **Loss**:
54
+
55
+ `sentence_transformers.losses.CosineSimilarityLoss.CosineSimilarityLoss`
56
+
57
+ Parameters of the fit()-Method:
58
+ ```
59
+ {
60
+ "epochs": 5,
61
+ "evaluation_steps": 1,
62
+ "evaluator": "sentence_transformers.evaluation.EmbeddingSimilarityEvaluator.EmbeddingSimilarityEvaluator",
63
+ "max_grad_norm": 1,
64
+ "optimizer_class": "<class 'transformers.optimization.AdamW'>",
65
+ "optimizer_params": {
66
+ "lr": 2e-05
67
+ },
68
+ "scheduler": "WarmupLinear",
69
+ "steps_per_epoch": null,
70
+ "warmup_steps": 6,
71
+ "weight_decay": 0.01
72
+ }
73
+ ```
74
+
75
+
76
+ ## Full Model Architecture
77
+ ```
78
+ SentenceTransformer(
79
+ (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: RobertaModel
80
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
81
+ (2): Normalize()
82
+ )
83
+ ```
84
+
85
+ ## Citing & Authors
86
+
87
+ <!--- Describe where people can find more information -->
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_all-distilroberta-v1/",
3
+ "architectures": [
4
+ "RobertaModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 3072,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "roberta",
19
+ "num_attention_heads": 12,
20
+ "num_hidden_layers": 6,
21
+ "pad_token_id": 1,
22
+ "position_embedding_type": "absolute",
23
+ "torch_dtype": "float32",
24
+ "transformers_version": "4.16.2",
25
+ "type_vocab_size": 1,
26
+ "use_cache": true,
27
+ "vocab_size": 50265
28
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "2.0.0",
4
+ "transformers": "4.6.1",
5
+ "pytorch": "1.8.1"
6
+ }
7
+ }
eval/similarity_evaluation_results.csv ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
2
+ 0,1,0.45892171995983294,0.5136906760016178,0.4480500006536557,0.5136906760016178,0.4538208185638512,0.49970404868474205,0.45892148035278574,0.5136906760016178
3
+ 0,2,0.4673271438681146,0.5136906760016178,0.45640708130214097,0.5136906760016178,0.4618807253038494,0.4857174213678663,0.4673272455984659,0.5136906760016178
4
+ 0,3,0.4852050267255112,0.5264057917442321,0.47153138326134353,0.5264057917442321,0.4783530796200351,0.5251342801699707,0.48520515342686993,0.5264057917442321
5
+ 0,4,0.49620613478444564,0.49588951396195774,0.48034859856136164,0.49588951396195774,0.48634903586114836,0.5136906760016178,0.49620604893580395,0.49588951396195774
6
+ 0,5,0.4675959646364591,0.42722788895184055,0.4476798737893579,0.42722788895184055,0.45007330440980264,0.44248602784297775,0.4675963628241339,0.42722788895184055
7
+ 0,6,0.44509795252180384,0.3712813796843376,0.4252970369839209,0.3712813796843376,0.4270528453186399,0.3712813796843376,0.4450975672917949,0.3712813796843376
8
+ 0,7,0.43515283963806484,0.37382440283286045,0.4145879089930638,0.37382440283286045,0.4133592539719618,0.345851148199109,0.4351529850882136,0.37382440283286045
9
+ 0,8,0.3931741848976815,0.31787789356535756,0.3648357958488576,0.31787789356535756,0.35481271324329383,0.31787789356535756,0.3931747678330642,0.31787789356535756
10
+ 0,9,0.3796623018239108,0.3318645208822333,0.34940321167501187,0.3318645208822333,0.344671087817085,0.28609010420882175,0.379662433910376,0.3318645208822333
11
+ 0,10,0.3983287063714711,0.3293214977337104,0.3693949709082921,0.3293214977337104,0.368663122636907,0.3216924282881418,0.3983290118307636,0.3293214977337104
12
+ 0,11,0.43126148265977543,0.3433081250505861,0.4053379824817655,0.3433081250505861,0.4107362110200949,0.35729475236746183,0.43126170508753126,0.3433081250505861
13
+ 0,-1,0.43126148265977543,0.3433081250505861,0.4053379824817655,0.3433081250505861,0.4107362110200949,0.35729475236746183,0.43126170508753126,0.3433081250505861
14
+ 1,1,0.4360188921268188,0.3433081250505861,0.41015666895796915,0.3433081250505861,0.4169944123964031,0.3827249838526905,0.4360187445388671,0.3433081250505861
15
+ 1,2,0.4540584174547239,0.4043406806151348,0.4307015089642643,0.4043406806151348,0.43985267449757903,0.37255289125859903,0.45405809330728564,0.4043406806151348
16
+ 1,3,0.44681482344446366,0.41196975006070335,0.422503671145419,0.41196975006070335,0.42908667412563783,0.38653951857547475,0.4468144378963373,0.41196975006070335
17
+ 1,4,0.4553935248758715,0.4056121921893962,0.4316349079835599,0.4056121921893962,0.4380392237695365,0.3839964954269519,0.45539327225647275,0.4056121921893962
18
+ 1,5,0.46313353721844075,0.4195988195062719,0.4424217262410759,0.4195988195062719,0.45094162218140477,0.42341335422905624,0.4631335428403658,0.4195988195062719
19
+ 1,6,0.4782651140736203,0.5098761412788335,0.4632941361619225,0.5098761412788335,0.47518099540746156,0.5098761412788335,0.47826505870420555,0.5098761412788335
20
+ 1,7,0.47544802497174876,0.546749976932415,0.4640806540241055,0.546749976932415,0.478190576256815,0.4933464908134349,0.47544785000513196,0.546749976932415
21
+ 1,8,0.46802225736860825,0.5429354422096307,0.45809972828504375,0.5429354422096307,0.4725983233689623,0.5187767222986636,0.46802247071080266,0.5429354422096307
22
+ 1,9,0.4549357223809874,0.46791625932820624,0.44406282811072884,0.46791625932820624,0.4576961876771825,0.4412145162687163,0.4549360119334626,0.46791625932820624
23
+ 1,10,0.4498024790284047,0.44884358571428484,0.440737812663459,0.44884358571428484,0.45407525634629625,0.4666447477539449,0.44980254158378397,0.44884358571428484
24
+ 1,11,0.4509230026798292,0.5276773033184935,0.4461088289004117,0.5276773033184935,0.4597768370301927,0.5289488148927549,0.45092306036979146,0.5276773033184935
25
+ 1,-1,0.4509230026798292,0.5276773033184935,0.4461088289004117,0.5276773033184935,0.4597768370301927,0.5289488148927549,0.45092306036979146,0.5276773033184935
26
+ 2,1,0.44709673337653244,0.5658226505463364,0.4448230595911736,0.5658226505463364,0.459440419553413,0.5658226505463364,0.44709665006712934,0.5658226505463364
27
+ 2,2,0.4439273056310685,0.5658226505463364,0.44406055940562594,0.5658226505463364,0.4594308389355642,0.5937959051800878,0.4439273000290313,0.5658226505463364
28
+ 2,3,0.4405511477652629,0.5937959051800878,0.4444528255196149,0.5937959051800878,0.46073015826071734,0.5937959051800878,0.44055121974287276,0.5937959051800878
29
+ 2,4,0.43220883040729124,0.5937959051800878,0.43582337580810504,0.5937959051800878,0.45269861763973923,0.5785377662889507,0.4322088006045344,0.5937959051800878
30
+ 2,5,0.4293962258806001,0.5658226505463364,0.4340225100633985,0.5658226505463364,0.4512180146263157,0.5785377662889507,0.42939647437329737,0.5658226505463364
31
+ 2,6,0.429497426660365,0.5658226505463364,0.43380077201148054,0.5658226505463364,0.4510522272693028,0.5670941621205978,0.42949759571943885,0.5658226505463364
32
+ 2,7,0.4266762501534375,0.5543790463779835,0.43206623137045996,0.5543790463779835,0.4493812569979816,0.5670941621205978,0.42667615215638116,0.5543790463779835
33
+ 2,8,0.42041991406855944,0.5327633496155393,0.4253818469071626,0.5327633496155393,0.44230013378384314,0.4895319560906506,0.4204200748501708,0.5327633496155393
34
+ 2,9,0.4138719347187233,0.4895319560906506,0.41709242809457364,0.4895319560906506,0.433091693431727,0.4895319560906506,0.4138721183874695,0.4895319560906506
35
+ 2,10,0.4080720198819035,0.4361284699716705,0.40927823679044134,0.4361284699716705,0.42413265094161223,0.41069823848644194,0.4080718855834293,0.4361284699716705
36
+ 2,11,0.4082085932423812,0.46155870145689915,0.40945035012625114,0.46155870145689915,0.4237663564300535,0.41069823848644194,0.40820838765857587,0.46155870145689915
37
+ 2,-1,0.4082085932423812,0.46155870145689915,0.40945035012625114,0.46155870145689915,0.4237663564300535,0.41069823848644194,0.40820838765857587,0.46155870145689915
38
+ 3,1,0.4091727035901019,0.43358544682314765,0.410166674720448,0.43358544682314765,0.4240417355075526,0.3827249838526905,0.40917265778122613,0.43358544682314765
39
+ 3,2,0.40619945410396785,0.43358544682314765,0.40620199382457767,0.43358544682314765,0.41965827766433134,0.3827249838526905,0.4061989439622096,0.43358544682314765
40
+ 3,3,0.4000484795377856,0.37255289125859903,0.39862609302984453,0.37255289125859903,0.41178227825676317,0.37255289125859903,0.4000478209616763,0.37255289125859903
41
+ 3,4,0.39506383228486586,0.37255289125859903,0.39326715677263446,0.37255289125859903,0.4060048542960496,0.35093719449615474,0.39506419181995583,0.37255289125859903
42
+ 3,5,0.39325009785865045,0.42341335422905624,0.3929109132480932,0.42341335422905624,0.40480904435680864,0.35093719449615474,0.3932502415387689,0.42341335422905624
43
+ 3,6,0.390142970433912,0.4513866088628077,0.3909848665414643,0.4513866088628077,0.4023433382760454,0.3763674259813833,0.3901433220202359,0.4513866088628077
44
+ 3,7,0.3898233197275008,0.4513866088628077,0.39184968331526543,0.4513866088628077,0.40339658365769154,0.43358544682314765,0.38982331923845925,0.4513866088628077
45
+ 3,8,0.39114078663148943,0.45774416673411483,0.39422155559123156,0.45774416673411483,0.40585014778149436,0.4361284699716705,0.3911407525228771,0.45774416673411483
46
+ 3,9,0.39247201421903166,0.45774416673411483,0.39633741086497193,0.45774416673411483,0.4083511282784633,0.4361284699716705,0.39247216614836217,0.45774416673411483
47
+ 3,10,0.39318391318179696,0.45774416673411483,0.39750092632114487,0.45774416673411483,0.4098799510860459,0.4361284699716705,0.39318377270397425,0.45774416673411483
48
+ 3,11,0.38981464605772553,0.4691877709024677,0.3934698895607529,0.4691877709024677,0.4069593003459497,0.4259563773775791,0.38981464263523063,0.4691877709024677
49
+ 3,-1,0.38981464605772553,0.4691877709024677,0.3934698895607529,0.4691877709024677,0.4069593003459497,0.4259563773775791,0.38981464263523063,0.4691877709024677
50
+ 4,1,0.3872810599641095,0.4691877709024677,0.3903310472411846,0.4691877709024677,0.40474448698348225,0.39798312274382763,0.38728151394494637,0.4691877709024677
51
+ 4,2,0.3860654851163914,0.4450290509915006,0.3884921345364612,0.4450290509915006,0.4041655351660248,0.39798312274382763,0.38606579738707164,0.4450290509915006
52
+ 4,3,0.38428976551180855,0.4450290509915006,0.3861272819475003,0.4450290509915006,0.4022884420953247,0.40179765746661195,0.38428967557696314,0.4450290509915006
53
+ 4,4,0.3812122299028358,0.4450290509915006,0.3823736697438846,0.4450290509915006,0.3987868075500073,0.40179765746661195,0.38121225179656865,0.4450290509915006
54
+ 4,5,0.37875469706796294,0.4450290509915006,0.3796391328529313,0.4450290509915006,0.39591902635212484,0.40179765746661195,0.37875453589052166,0.4450290509915006
55
+ 4,6,0.37645491328584235,0.4450290509915006,0.377424594209919,0.4450290509915006,0.3937943913342341,0.40179765746661195,0.3764547912681717,0.4450290509915006
56
+ 4,7,0.37462280362374656,0.4450290509915006,0.3754291660603055,0.4450290509915006,0.39196981061937824,0.40179765746661195,0.37462254235427905,0.4450290509915006
57
+ 4,8,0.37326743157189074,0.4450290509915006,0.37409883951281364,0.4450290509915006,0.39077440755871545,0.40179765746661195,0.37326735024676394,0.4450290509915006
58
+ 4,9,0.37249660023000075,0.4450290509915006,0.37347105947734077,0.4450290509915006,0.3900649213758474,0.40179765746661195,0.37249633277017347,0.4450290509915006
59
+ 4,10,0.37222575714604805,0.4450290509915006,0.37334811969117404,0.4450290509915006,0.38982815409809474,0.39798312274382763,0.37222573984048873,0.4450290509915006
60
+ 4,11,0.3726087306044415,0.4450290509915006,0.37385814314501054,0.4450290509915006,0.3902934979208738,0.39798312274382763,0.3726084165143944,0.4450290509915006
61
+ 4,-1,0.3726087306044415,0.4450290509915006,0.37385814314501054,0.4450290509915006,0.3902934979208738,0.39798312274382763,0.3726084165143944,0.4450290509915006
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
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
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.models.Normalize"
19
+ }
20
+ ]
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a6a573653564e2689e7cc9120b0b5ce8521660e3e06e45ea72a67c8186b2e15
3
+ size 328517361
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }
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.7534333699670948,0.34862028508267734,0.7344290336184773,0.34862028508267734,0.732401418683328,0.36200575638800453,0.7534333195675238,0.34855300443390613
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "pad_token": "<pad>", "cls_token": "<s>", "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false}}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "add_prefix_space": false, "errors": "replace", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": "<mask>", "trim_offsets": true, "model_max_length": 512, "special_tokens_map_file": null, "name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_all-distilroberta-v1/", "tokenizer_class": "RobertaTokenizer"}
vocab.json ADDED
The diff for this file is too large to render. See raw diff