Upload 13 files

1_Pooling/config.json

@@ -0,0 +1,10 @@
+{
+  "word_embedding_dimension": 384,
+  "pooling_mode_cls_token": false,
+  "pooling_mode_mean_tokens": true,
+  "pooling_mode_max_tokens": false,
+  "pooling_mode_mean_sqrt_len_tokens": false,
+  "pooling_mode_weightedmean_tokens": false,
+  "pooling_mode_lasttoken": false,
+  "include_prompt": true
+}

README.md

@@ -0,0 +1,128 @@
+---
+library_name: sentence-transformers
+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 384 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 137553 with parameters:
+```
+{'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+```
+
+**Loss**:
+
+`sentence_transformers.losses.MSELoss.MSELoss` 
+
+Parameters of the fit()-Method:
+```
+{
+    "epochs": 1,
+    "evaluation_steps": 5000,
+    "evaluator": "sentence_transformers.evaluation.SequentialEvaluator.SequentialEvaluator",
+    "max_grad_norm": 1,
+    "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
+    "optimizer_params": {
+        "eps": 1e-06,
+        "lr": 0.0001
+    },
+    "scheduler": "WarmupLinear",
+    "steps_per_epoch": null,
+    "warmup_steps": 1000,
+    "weight_decay": 0.01
+}
+```
+
+
+## Full Model Architecture
+```
+SentenceTransformer(
+  (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: BertModel 
+  (1): Pooling({'word_embedding_dimension': 384, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
+)
+```
+
+## Citing & Authors
+
+<!--- Describe where people can find more information -->

config.json

@@ -0,0 +1,25 @@
+{
+  "_name_or_path": "TaylorAI/gte-tiny",
+  "architectures": [
+    "BertModel"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 384,
+  "initializer_range": 0.02,
+  "intermediate_size": 1536,
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 3,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "torch_dtype": "float32",
+  "transformers_version": "4.40.0",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 30522
+}

config_sentence_transformers.json

@@ -0,0 +1,9 @@
+{
+  "__version__": {
+    "sentence_transformers": "2.2.2",
+    "transformers": "4.34.0",
+    "pytorch": "2.0.1+cu118"
+  },
+  "prompts": {},
+  "default_prompt_name": null
+}

eval/mse_evaluation__results.csv

@@ -0,0 +1,29 @@
+epoch,steps,MSE
+0,5000,0.47166235744953156
+0,10000,0.37070035468786955
+0,15000,0.32915491610765457
+0,20000,0.3022111253812909
+0,25000,0.2860546577721834
+0,30000,0.27226749807596207
+0,35000,0.2621056279167533
+0,40000,0.25275147054344416
+0,45000,0.24559586308896542
+0,50000,0.2409756649285555
+0,55000,0.23480404634028673
+0,60000,0.23111277259886265
+0,65000,0.2271266421303153
+0,70000,0.22308838088065386
+0,75000,0.2202761359512806
+0,80000,0.21781811956316233
+0,85000,0.21458840928971767
+0,90000,0.21309000439941883
+0,95000,0.2095935633406043
+0,100000,0.20842656958848238
+0,105000,0.20773126743733883
+0,110000,0.2059056656435132
+0,115000,0.20338515751063824
+0,120000,0.20260869059711695
+0,125000,0.2014710335060954
+0,130000,0.20016569178551435
+0,135000,0.199802010320127
+0,-1,0.19947525579482317

eval/similarity_evaluation_sts-dev_results.csv

@@ -0,0 +1,29 @@
+epoch,steps,cosine_pearson,cosine_spearman,euclidean_pearson,euclidean_spearman,manhattan_pearson,manhattan_spearman,dot_pearson,dot_spearman
+0,5000,0.8482784841781801,0.8524673285103249,0.8527363303816113,0.8536974318032373,0.8507429053520061,0.8518087596830178,0.7290294796138388,0.7222457751199094
+0,10000,0.851522289059487,0.8550575570365678,0.8561895858778579,0.8564992433428477,0.8549097711624272,0.8553545699560491,0.7413520865451514,0.7373012959517407
+0,15000,0.853710809631439,0.8565147992750579,0.8577158540045073,0.8577098527181514,0.8565144787975028,0.8568546919864072,0.7504689821086896,0.7449558272692272
+0,20000,0.8540319978452439,0.8569741180973768,0.8578535262979087,0.8580078785972628,0.8565983775145072,0.8570460462108194,0.7479221251875051,0.7434152209949655
+0,25000,0.8546565882275716,0.8578572076747778,0.8590097969084035,0.8589626258235827,0.8578829955242684,0.8580656447783175,0.7465150443680822,0.7418695315450354
+0,30000,0.8566672105901525,0.8594213535205055,0.8602676272434086,0.8602408135372777,0.8590713876871994,0.8593383146946152,0.7526094791593237,0.748224769026938
+0,35000,0.8571100252746132,0.8600871771852339,0.8608204232615566,0.8610102144234375,0.8595434622508469,0.8599233613611874,0.753956316112839,0.7502490202011034
+0,40000,0.8559139037907579,0.8588256318306422,0.859604173569988,0.8597011761546718,0.8584894417578237,0.8587051292048292,0.7474515413384551,0.741869366064494
+0,45000,0.856265008742947,0.8592787334409433,0.8601104783499282,0.8601181309432502,0.8592395715659137,0.8594094204897488,0.7484793142063589,0.7435075575864867
+0,50000,0.8569211165237697,0.8598292756363145,0.8605543442052597,0.8606103458204372,0.8595676020012094,0.8597390195971244,0.7515427958024276,0.7472764189537775
+0,55000,0.8562075563924645,0.8589091799310042,0.859841781260059,0.8598799538579605,0.8587882409340232,0.8590060600172196,0.7475246493261491,0.7428149005248468
+0,60000,0.8569707723246641,0.8595952806871071,0.860566323618586,0.8604651304178604,0.8596098187861387,0.8597964769320989,0.7535645460750052,0.7486540573552126
+0,65000,0.8569103452131722,0.8600268782112365,0.8614217305429412,0.8608495424800336,0.8603240057571976,0.8602177921690977,0.7460511702193234,0.7402588649530701
+0,70000,0.8573467610257368,0.8598835247841208,0.8610375445882468,0.861163641575446,0.8600034121583955,0.8602742993254814,0.7495966667469552,0.7445113463075477
+0,75000,0.8569202261382182,0.8596530950351353,0.8607016007767625,0.8604840379042015,0.8596498923735942,0.8596676423237303,0.7515624025047947,0.7468403686053563
+0,80000,0.8576730599037291,0.8604011756225682,0.8615432661698413,0.8612236797489058,0.8605588884907639,0.8603870894388059,0.750107686339764,0.7452585641329533
+0,85000,0.8583390040790053,0.8611601059858207,0.8621337513282632,0.8619397257402335,0.861063106730213,0.8609504973004285,0.7546225161201876,0.7497468572659596
+0,90000,0.8575744131143879,0.8607364409784936,0.8615518120348374,0.8614321310843385,0.8605467563725214,0.8605709159532309,0.7542955531758224,0.7494734075655113
+0,95000,0.8581535214603767,0.8608581047635226,0.8614857625667244,0.861603823373491,0.8604308998084226,0.8605555341452761,0.7562258408218162,0.7519859032951348
+0,100000,0.8575272792375968,0.8604531151601916,0.8616026317998216,0.861469841840493,0.8607229105317097,0.8604864062332358,0.7509198586787145,0.7465743710908277
+0,105000,0.8575445480655535,0.8604985030918701,0.8615044926265819,0.8612746594455564,0.8605518068670269,0.8605125841872215,0.7522032211150207,0.7479008962525083
+0,110000,0.8576966144385598,0.8605407451139044,0.8614088201804478,0.8614305510120751,0.8603840346381387,0.8605051251073474,0.7517030043173099,0.7470494201254125
+0,115000,0.857477524915541,0.8604521454086326,0.8612827456575336,0.8611852366618776,0.8603034722973986,0.8603012520660652,0.7558904929775052,0.7514950677688355
+0,120000,0.8577417338422507,0.8606742469853947,0.8615645142474043,0.861536146156713,0.8604828695956938,0.8605137336542058,0.7514754637331444,0.7470229770466421
+0,125000,0.8579218216815234,0.860733086883652,0.8618307953661968,0.8615596098198863,0.8608479412078883,0.8606167594012378,0.7522352230909667,0.7479105813125642
+0,130000,0.858032479066429,0.8608713662140498,0.8618483310090321,0.8616822014176498,0.8609555683008904,0.8608600630743983,0.7529330321732546,0.748948000177042
+0,135000,0.8578263948202207,0.860584293186662,0.8617266127627646,0.8614490090859027,0.8607919607942601,0.8606299472574345,0.7530837013484516,0.7488705563911164
+0,-1,0.8578309282683841,0.8606207432650688,0.8617281651065495,0.86147026990066,0.8607882122917774,0.8605566340396457,0.7527337207540649,0.7485823749213384

model.safetensors

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

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"
+  }
+]

sentence_bert_config.json

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

special_tokens_map.json

@@ -0,0 +1,44 @@
+{
+  "additional_special_tokens": [
+    "[PAD]",
+    "[UNK]",
+    "[CLS]",
+    "[SEP]",
+    "[MASK]"
+  ],
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "[MASK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "[PAD]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,71 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "[PAD]",
+    "[UNK]",
+    "[CLS]",
+    "[SEP]",
+    "[MASK]"
+  ],
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "max_length": 128,
+  "model_max_length": 512,
+  "never_split": null,
+  "pad_to_multiple_of": null,
+  "pad_token": "[PAD]",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "[SEP]",
+  "stride": 0,
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "[UNK]"
+}