Model save

README.md

@@ -1,32 +1,14 @@
 ---
 library_name: transformers
-language:
-- en
 base_model: gokulsrinivasagan/bert_base_lda_50_v1
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - matthews_correlation
 - accuracy
 model-index:
 - name: bert_base_lda_50_v1_cola
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: GLUE COLA
-      type: glue
-      args: cola
-    metrics:
-    - name: Matthews Correlation
-      type: matthews_correlation
-      value: 0.0
-    - name: Accuracy
-      type: accuracy
-      value: 0.6912751793861389
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert_base_lda_50_v1_cola
 
-This model is a fine-tuned version of [gokulsrinivasagan/bert_base_lda_50_v1](https://huggingface.co/gokulsrinivasagan/bert_base_lda_50_v1) on the GLUE COLA dataset.
+This model is a fine-tuned version of [gokulsrinivasagan/bert_base_lda_50_v1](https://huggingface.co/gokulsrinivasagan/bert_base_lda_50_v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6180
-- Matthews Correlation: 0.0
-- Accuracy: 0.6913
+- Loss: 0.6751
+- Matthews Correlation: 0.1011
+- Accuracy: 0.6529
 
 ## Model description
 
@@ -57,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
@@ -69,13 +51,15 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Matthews Correlation | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------------------:|:--------:|
-| 0.9041        | 1.0   | 34   | 0.6207          | 0.0                  | 0.6913   |
-| 0.6134        | 2.0   | 68   | 0.6180          | 0.0                  | 0.6913   |
-| 0.6096        | 3.0   | 102  | 0.6192          | 0.0                  | 0.6913   |
-| 0.6106        | 4.0   | 136  | 0.6189          | 0.0                  | 0.6913   |
-| 0.6105        | 5.0   | 170  | 0.6192          | 0.0                  | 0.6913   |
-| 0.609         | 6.0   | 204  | 0.6193          | 0.0                  | 0.6913   |
-| 0.6089        | 7.0   | 238  | 0.6189          | 0.0                  | 0.6913   |
+| 0.6197        | 1.0   | 34   | 0.6198          | 0.0                  | 0.6913   |
+| 0.6125        | 2.0   | 68   | 0.6179          | 0.0                  | 0.6913   |
+| 0.6104        | 3.0   | 102  | 0.6171          | 0.0                  | 0.6913   |
+| 0.6101        | 4.0   | 136  | 0.6144          | 0.0                  | 0.6913   |
+| 0.5982        | 5.0   | 170  | 0.6144          | 0.0                  | 0.6913   |
+| 0.5682        | 6.0   | 204  | 0.6281          | 0.0526               | 0.6874   |
+| 0.5152        | 7.0   | 238  | 0.6455          | 0.1003               | 0.6568   |
+| 0.4552        | 8.0   | 272  | 0.6827          | 0.0970               | 0.6635   |
+| 0.4108        | 9.0   | 306  | 0.6751          | 0.1011               | 0.6529   |
 
 
 ### Framework versions

logs/events.out.tfevents.1733309148.ki-g0008.1207984.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:24fa4de0e3b15edbc15749df376278744f263f4576f090a8c53cbc0c77d86ec5
-size 5672
+oid sha256:15dab4765d9899a1494388dfb4b63ef640421d9d32acd073e97f78ae2bd47dd7
+size 10788

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:83f49678cacaa0c5eea4d1c1795506130377d4b6d249e19d611798c6cf3cd48a
+oid sha256:a48bce4494ec6e8c8f6d33fe21edad517f98d5fb4d4e9f81576206c76430482e
 size 437953248