Model save

README.md

@@ -1,28 +1,13 @@
 ---
 library_name: transformers
-language:
-- en
 base_model: gokulsrinivasagan/distilbert_lda_5_v1
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - spearmanr
 model-index:
 - name: distilbert_lda_5_v1_stsb
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: GLUE STSB
-      type: glue
-      args: stsb
-    metrics:
-    - name: Spearmanr
-      type: spearmanr
-      value: .nan
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert_lda_5_v1_stsb
 
-This model is a fine-tuned version of [gokulsrinivasagan/distilbert_lda_5_v1](https://huggingface.co/gokulsrinivasagan/distilbert_lda_5_v1) on the GLUE STSB dataset.
+This model is a fine-tuned version of [gokulsrinivasagan/distilbert_lda_5_v1](https://huggingface.co/gokulsrinivasagan/distilbert_lda_5_v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.2568
-- Pearson: nan
-- Spearmanr: nan
-- Combined Score: nan
+- Loss: 1.0893
+- Pearson: 0.7716
+- Spearmanr: 0.7719
+- Combined Score: 0.7718
 
 ## Model description
 
@@ -54,7 +39,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.001
+- learning_rate: 5e-05
 - train_batch_size: 256
 - eval_batch_size: 256
 - seed: 10
@@ -66,12 +51,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Pearson | Spearmanr | Combined Score |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:---------:|:--------------:|
-| 6.3938        | 1.0   | 23   | 2.2568          | nan     | nan       | nan            |
-| 2.2292        | 2.0   | 46   | 2.2730          | nan     | nan       | nan            |
-| 2.2087        | 3.0   | 69   | 2.3180          | -0.0301 | -0.0313   | -0.0307        |
-| 2.2043        | 4.0   | 92   | 2.6549          | 0.0080  | 0.0111    | 0.0095         |
-| 2.2273        | 5.0   | 115  | 2.6159          | 0.0018  | 0.0021    | 0.0019         |
-| 2.1946        | 6.0   | 138  | 2.3650          | nan     | nan       | nan            |
+| 2.4726        | 1.0   | 23   | 1.7690          | 0.5073  | 0.5038    | 0.5056         |
+| 1.3012        | 2.0   | 46   | 1.3896          | 0.7004  | 0.7125    | 0.7064         |
+| 0.8284        | 3.0   | 69   | 0.9456          | 0.7721  | 0.7714    | 0.7718         |
+| 0.5881        | 4.0   | 92   | 0.9819          | 0.7670  | 0.7671    | 0.7670         |
+| 0.4537        | 5.0   | 115  | 0.9149          | 0.7745  | 0.7714    | 0.7730         |
+| 0.3479        | 6.0   | 138  | 1.1078          | 0.7723  | 0.7729    | 0.7726         |
+| 0.2723        | 7.0   | 161  | 0.9508          | 0.7747  | 0.7725    | 0.7736         |
+| 0.2507        | 8.0   | 184  | 0.9537          | 0.7685  | 0.7651    | 0.7668         |
+| 0.2101        | 9.0   | 207  | 1.0716          | 0.7639  | 0.7611    | 0.7625         |
+| 0.1713        | 10.0  | 230  | 1.0893          | 0.7716  | 0.7719    | 0.7718         |
 
 
 ### Framework versions

logs/events.out.tfevents.1733313849.ki-g0008.1206098.12

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0de93c22ffb6c51d7c5a9e4ac2220de90d788cab08e5e43d000f183babdc9090
-size 10833
+oid sha256:00ad0bd3f5712de82928ecc51f3c9f4a07eaf886dd65fc55ebf9617c971c7633
+size 11831

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2767ef76a61c6804e39bd9326fddd1bc40a1a23833b095a56b9d25267edad6cd
+oid sha256:abb6bf93f1a6aa142a7006482b329c8cf2193fff7ae150e21d7180893bb89eb8
 size 267829484