MrezaPRZ/sql-similarity-predictor

README.md

@@ -0,0 +1,69 @@
+---
+license: other
+base_model: deepseek-ai/deepseek-coder-1.3b-instruct
+tags:
+- generated_from_trainer
+model-index:
+- name: encoder
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# encoder
+
+This model is a fine-tuned version of [deepseek-ai/deepseek-coder-1.3b-instruct](https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-instruct) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0670
+- Mse: 1.0575
+- Rmse: 1.0283
+- Mae: 0.9076
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 12
+- eval_batch_size: 8
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.01
+- num_epochs: 2
+
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Mse    | Rmse   | Mae    |
+|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|
+| 0.0988        | 0.21  | 3000  | 0.0928          | 1.4795 | 1.2164 | 1.1099 |
+| 0.0793        | 0.42  | 6000  | 0.0867          | 1.4370 | 1.1987 | 1.1076 |
+| 0.0702        | 0.63  | 9000  | 0.0777          | 0.7554 | 0.8691 | 0.7701 |
+| 0.0634        | 0.84  | 12000 | 0.0716          | 1.0950 | 1.0464 | 0.9449 |
+| 0.0563        | 1.05  | 15000 | 0.0686          | 0.9966 | 0.9983 | 0.8899 |
+| 0.0484        | 1.26  | 18000 | 0.0673          | 1.0653 | 1.0321 | 0.9161 |
+| 0.0466        | 1.47  | 21000 | 0.0671          | 1.0877 | 1.0429 | 0.9219 |
+| 0.0462        | 1.68  | 24000 | 0.0670          | 1.0613 | 1.0302 | 0.9090 |
+| 0.046         | 1.89  | 27000 | 0.0670          | 1.0575 | 1.0283 | 0.9076 |
+
+
+### Framework versions
+
+- Transformers 4.37.2
+- Pytorch 2.1.0.dev20230605+cu121
+- Datasets 2.17.0
+- Tokenizers 0.15.2

config.json

@@ -0,0 +1,38 @@
+{
+  "_name_or_path": "deepseek-ai/deepseek-coder-1.3b-instruct",
+  "architectures": [
+    "LlamaForSequenceClassification"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 32013,
+  "eos_token_id": 32021,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
+  "id2label": {
+    "0": "LABEL_0"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 5504,
+  "label2id": {
+    "LABEL_0": 0
+  },
+  "max_position_embeddings": 16384,
+  "model_type": "llama",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "num_key_value_heads": 16,
+  "pad_token_id": 32014,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": {
+    "factor": 4.0,
+    "type": "linear"
+  },
+  "rope_theta": 100000,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.37.2",
+  "use_cache": false,
+  "vocab_size": 32256
+}

model.safetensors

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

tmp-checkpoint-40/config.json

@@ -0,0 +1,38 @@
+{
+  "_name_or_path": "deepseek-ai/deepseek-coder-1.3b-instruct",
+  "architectures": [
+    "LlamaForSequenceClassification"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 32013,
+  "eos_token_id": 32021,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
+  "id2label": {
+    "0": "LABEL_0"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 5504,
+  "label2id": {
+    "LABEL_0": 0
+  },
+  "max_position_embeddings": 16384,
+  "model_type": "llama",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "num_key_value_heads": 16,
+  "pad_token_id": 32014,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": {
+    "factor": 4.0,
+    "type": "linear"
+  },
+  "rope_theta": 100000,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.37.2",
+  "use_cache": false,
+  "vocab_size": 32256
+}

tmp-checkpoint-40/model.safetensors

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

tmp-checkpoint-40/optimizer.pt

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

tmp-checkpoint-40/training_args.bin

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

training_args.bin

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