init

1_Pooling/config.json

@@ -0,0 +1,7 @@
+{
+  "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
+}

README.md

@@ -0,0 +1,125 @@
+---
+pipeline_tag: sentence-similarity
+tags:
+- sentence-transformers
+- feature-extraction
+- sentence-similarity
+- transformers
+---
+
+# {MODEL_NAME}
+
+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.
+
+<!--- Describe your model here -->
+
+## Usage (Sentence-Transformers)
+
+Using this model becomes easy when you have [sentence-transformers](https://www.SBERT.net) installed:
+
+```
+pip install -U sentence-transformers
+```
+
+Then you can use the model like this:
+
+```python
+from sentence_transformers import SentenceTransformer
+sentences = ["This is an example sentence", "Each sentence is converted"]
+
+model = SentenceTransformer('{MODEL_NAME}')
+embeddings = model.encode(sentences)
+print(embeddings)
+```
+
+
+
+## Usage (HuggingFace Transformers)
+Without [sentence-transformers](https://www.SBERT.net), you can use the model like this: First, you pass your input through the transformer model, then you have to apply the right pooling-operation on-top of the contextualized word embeddings.
+
+```python
+from transformers import AutoTokenizer, AutoModel
+import torch
+
+
+#Mean Pooling - Take attention mask into account for correct averaging
+def mean_pooling(model_output, attention_mask):
+    token_embeddings = model_output[0] #First element of model_output contains all token embeddings
+    input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
+    return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
+
+
+# Sentences we want sentence embeddings for
+sentences = ['This is an example sentence', 'Each sentence is converted']
+
+# Load model from HuggingFace Hub
+tokenizer = AutoTokenizer.from_pretrained('{MODEL_NAME}')
+model = AutoModel.from_pretrained('{MODEL_NAME}')
+
+# Tokenize sentences
+encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
+
+# Compute token embeddings
+with torch.no_grad():
+    model_output = model(**encoded_input)
+
+# Perform pooling. In this case, mean pooling.
+sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
+
+print("Sentence embeddings:")
+print(sentence_embeddings)
+```
+
+
+
+## Evaluation Results
+
+<!--- Describe how your model was evaluated -->
+
+For an automated evaluation of this model, see the *Sentence Embeddings Benchmark*: [https://seb.sbert.net](https://seb.sbert.net?model_name={MODEL_NAME})
+
+
+## Training
+The model was trained with the parameters:
+
+**DataLoader**:
+
+`torch.utils.data.dataloader.DataLoader` of length 136 with parameters:
+```
+{'batch_size': 128, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+```
+
+**Loss**:
+
+`sentence_transformers.losses.CosineSimilarityLoss.CosineSimilarityLoss` 
+
+Parameters of the fit()-Method:
+```
+{
+    "epochs": 4,
+    "evaluation_steps": 13,
+    "evaluator": "sentence_transformers.evaluation.EmbeddingSimilarityEvaluator.EmbeddingSimilarityEvaluator",
+    "max_grad_norm": 1,
+    "optimizer_class": "<class 'transformers.optimization.AdamW'>",
+    "optimizer_params": {
+        "lr": 2e-05
+    },
+    "scheduler": "WarmupLinear",
+    "steps_per_epoch": null,
+    "warmup_steps": 14,
+    "weight_decay": 0.01
+}
+```
+
+
+## Full Model Architecture
+```
+SentenceTransformer(
+  (0): Transformer({'max_seq_length': 64, 'do_lower_case': False}) with Transformer model: RobertaModel 
+  (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})
+)
+```
+
+## Citing & Authors
+
+<!--- Describe where people can find more information -->

config.json

@@ -0,0 +1,29 @@
+{
+  "_name_or_path": "2-nli",
+  "architectures": [
+    "RobertaModel"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "classifier_dropout": null,
+  "eos_token_id": 2,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 514,
+  "model_type": "roberta",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "position_embedding_type": "absolute",
+  "tokenizer_class": "BertTokenizer",
+  "torch_dtype": "float32",
+  "transformers_version": "4.19.2",
+  "type_vocab_size": 1,
+  "use_cache": true,
+  "vocab_size": 32000
+}

config_sentence_transformers.json

@@ -0,0 +1,7 @@
+{
+  "__version__": {
+    "sentence_transformers": "2.2.0",
+    "transformers": "4.19.2",
+    "pytorch": "1.10.1+cu111"
+  }
+}

eval/similarity_evaluation_sts-dev_results.csv

@@ -0,0 +1,45 @@
+epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
+0,13,0.840173574575316,0.838095322331488,0.8286651385659133,0.8320831116685604,0.8287316783198648,0.8320954659113529,0.8209369470153028,0.8156436890315599
+0,26,0.8371863153814791,0.8338717217514832,0.8205183077847656,0.8270253068566726,0.8206636068966722,0.8270476412143913,0.8218082025702487,0.8181354912657327
+0,39,0.8494196987075502,0.8459726869104756,0.8361844381385505,0.8411977770762317,0.8362236808273045,0.8411626083236909,0.8338617392651451,0.8285164702806552
+0,52,0.8539708021623097,0.8510609737169117,0.8440597498936332,0.8486884760958173,0.8439456941289454,0.8484418634396012,0.8353826995550104,0.8296580556012464
+0,65,0.857034797635417,0.8546809474282939,0.8471566068514274,0.852076717339059,0.8470150168083339,0.8517953831470374,0.8372326752968932,0.8319800503562287
+0,78,0.8570350198661547,0.8548685858784162,0.8478859320899844,0.8522973837631964,0.8476663072888273,0.8519467699625926,0.8360678725387414,0.8304435959976367
+0,91,0.858982605293718,0.856630837200268,0.8481207766200408,0.8524079696287328,0.8479040508241764,0.8522247232960709,0.8397536762904672,0.8345044420150022
+0,104,0.8622011175531357,0.8598668454252475,0.8512910025315441,0.8557607236653706,0.8511199333972042,0.8555213819664478,0.8428182501006237,0.8378095143717595
+0,117,0.8631322488903511,0.8609592791176339,0.8513824873225329,0.8561050283944893,0.8512353982550677,0.8558104471660098,0.843604642067139,0.8391355369435497
+0,130,0.8629620948497575,0.8612828373581486,0.852445494971362,0.8565417854615515,0.8523066016699872,0.8563477123982908,0.8427909542486722,0.8381035930381442
+0,-1,0.8649553267658603,0.8631446689809993,0.8529231278104039,0.8581811171513349,0.8527000944780232,0.8579245305141419,0.8450121761999848,0.8406888848384645
+1,13,0.8652365049435421,0.8634720008606691,0.8530556684685462,0.8594485248278001,0.8528866218151859,0.8591538922367353,0.8466106282903358,0.8426839978732009
+1,26,0.8658221870903494,0.8642291924403861,0.8555922319515743,0.8613499381091321,0.8553714096939835,0.8610881756162121,0.8471802043805328,0.8427752531033437
+1,39,0.8640849435779709,0.8631426839423925,0.8543341057098628,0.8598275566997508,0.854131783068022,0.8596191082004716,0.8455129809762828,0.8414975693957001
+1,52,0.8647356768324745,0.8635269098589013,0.8554223788000532,0.8601752457467844,0.8552987468897215,0.8599974141975133,0.8462126262361407,0.8416740431237697
+1,65,0.8655821317527608,0.8645587653166551,0.8553563714259382,0.8608527650240532,0.8551425731960431,0.8605676347407091,0.8464758247211841,0.8425899194941123
+1,78,0.8663588675931928,0.865139228039893,0.8548398829304542,0.8608589967970643,0.8546129659754933,0.8605498984016274,0.8488943602361901,0.8449484398036953
+1,91,0.8664113286502602,0.8649767609519097,0.854892738711776,0.8610412544834212,0.8546991354495329,0.8607656175535816,0.8483163864834093,0.8443587550395061
+1,104,0.8650689590005896,0.863699007220169,0.853739446844429,0.8593942020851261,0.8535766806865603,0.8590228569129204,0.8470059732599704,0.8430295060260515
+1,117,0.8665353934940071,0.8655195689879606,0.8563919180271475,0.8616608609430584,0.8562184790756451,0.8614047281033689,0.8487200126909797,0.8444767870254865
+1,130,0.8664300690737297,0.865571184540681,0.8566064812934235,0.8622698787194275,0.8564341714044499,0.8620479459290493,0.848061641921869,0.8443784753662604
+1,-1,0.8661316603825252,0.8653281863876557,0.855974355132375,0.8617937120220726,0.855819750303604,0.8615451390072588,0.8482242785732571,0.8447189993441905
+2,13,0.866747854861505,0.8661356980991812,0.8554926149864794,0.8616609175182531,0.8553039533899902,0.8614334096251158,0.8503116213796384,0.8473136176565894
+2,26,0.8671871494707486,0.8659632785633392,0.8563965476688629,0.8619833091462822,0.8561564817437395,0.8616585962291827,0.8499167311251462,0.8460167277130445
+2,39,0.86664099590433,0.8657897810986168,0.8567740626433016,0.8623064800918516,0.8565247587400275,0.8619395571787992,0.8482763772502576,0.8445900709948548
+2,52,0.8661675458410681,0.8654086897114864,0.855300918937258,0.8614137784115875,0.8550614873241764,0.8610775391535519,0.84806702154158,0.8448155550112744
+2,65,0.8667137908801402,0.8657605442941214,0.8564085235807335,0.8620745650336229,0.8561369105364691,0.8616842485299234,0.8484654533845171,0.8446590304749071
+2,78,0.8667887021729552,0.8657663258563987,0.8571590554412276,0.8620679123860784,0.8568771984862733,0.8617261439955124,0.8474590652558861,0.8429897061242195
+2,91,0.8670032031443542,0.866119463887954,0.8562101578489287,0.8619238228939545,0.85595873082041,0.8615602331273123,0.8480500913392531,0.8442938371464792
+2,104,0.8678913054991113,0.8673299130933578,0.8577133769428058,0.8635122841527879,0.8574153221304429,0.8631973618336097,0.8494425222608484,0.8458298695089989
+2,117,0.8671358053645019,0.8666280822514199,0.8574714220044806,0.8630397601923891,0.8572255231102923,0.8627128342189467,0.8486385992949529,0.8449187712290472
+2,130,0.867297914835443,0.8666708561512251,0.8576578714178784,0.86301894445677,0.8574616044121085,0.862773208687516,0.848884603769172,0.8448544738055712
+2,-1,0.8673916283148887,0.8667764112654047,0.8579334843259333,0.86316511720213,0.8577385353887954,0.8629299410955781,0.8486762921510912,0.8445438233788372
+3,13,0.8670817541775118,0.8662833196729459,0.8566175337791606,0.8623143480256884,0.8564237972397568,0.862079433503667,0.8477453118466537,0.8439407004505581
+3,26,0.8669752641067601,0.8663749262408468,0.8573202942110567,0.8627991150325515,0.8571217594075893,0.8624854125899885,0.8476216487291267,0.8436838513150303
+3,39,0.8669746477050544,0.866451197133775,0.8573929864214485,0.862911563426453,0.8571679334029205,0.8625582838901306,0.8479421142878307,0.8441425996596054
+3,52,0.8674552475904721,0.8669044335087338,0.8575009702790937,0.8631725746180012,0.8572538195675206,0.8628316715389747,0.8484784196502632,0.8448256474303713
+3,65,0.8677479322415691,0.8673093544730277,0.857652339469569,0.8633947220975304,0.8573923188441335,0.8630705676820345,0.8491329361817883,0.8455323071909955
+3,78,0.8676935016753708,0.867281290598116,0.8577173125139128,0.8634252883172676,0.8574606251272271,0.8631007502019445,0.8494149123516681,0.8458176608086381
+3,91,0.8676466999858835,0.8671367312954688,0.8575034335532818,0.8631833810083913,0.8572476403003226,0.8629042768662577,0.8491850838514424,0.845538083718587
+3,104,0.8676624020164339,0.8670982131310448,0.8574798490713721,0.8631678092802401,0.8572302117472728,0.862856464463703,0.8489483728789051,0.8452803843409546
+3,117,0.8677311985204097,0.8671904844038181,0.8574431594473895,0.8632114180015596,0.8571926674652525,0.8629112047461654,0.8489009994021832,0.8452732904358602
+3,130,0.867738471211877,0.8671813679069309,0.857403175257186,0.8631973673735163,0.8571553268250389,0.8628478010674976,0.8488687152552925,0.8452674167102061
+3,-1,0.8677407616603636,0.8671897071391101,0.8574052391314121,0.8632041920637725,0.857158187185046,0.8628517402577571,0.8488743873752991,0.8452762718553655

modules.json

@@ -0,0 +1,14 @@
+[
+  {
+    "idx": 0,
+    "name": "0",
+    "path": "",
+    "type": "sentence_transformers.models.Transformer"
+  },
+  {
+    "idx": 1,
+    "name": "1",
+    "path": "1_Pooling",
+    "type": "sentence_transformers.models.Pooling"
+  }
+]

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a8a5a18745d36e491ec679dc2c8085d3da1ea71f7c2f16fdbe5cd65caf1632b
+size 442541937

sentence_bert_config.json

@@ -0,0 +1,4 @@
+{
+  "max_seq_length": 64,
+  "do_lower_case": false
+}

similarity_evaluation_sts-dev_results.csv

@@ -0,0 +1,2 @@
+epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
+-1,-1,0.8678913054991113,0.8673299130933578,0.8577133769428058,0.8635122841527879,0.8574153221304429,0.8631973618336097,0.8494425222608484,0.8458298695089989

special_tokens_map.json

@@ -0,0 +1 @@
+{"bos_token": "[CLS]", "eos_token": "[SEP]", "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}

tokenizer_config.json

@@ -0,0 +1 @@
+{"do_lower_case": false, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "tokenize_chinese_chars": true, "strip_accents": null, "do_basic_tokenize": true, "never_split": null, "bos_token": "[CLS]", "eos_token": "[SEP]", "model_max_length": 512, "special_tokens_map_file": "/home/cleaning/.cache/huggingface/transformers/9d0c87e44b00acfbfbae931b2e4068eb6311a0c3e71e23e5400bdf57cab4bfbf.70c17d6e4d492c8f24f5bb97ab56c7f272e947112c6faf9dd846da42ba13eb23", "name_or_path": "2-nli", "tokenizer_class": "BertTokenizer"}