Model save

README.md

@@ -1,32 +1,14 @@
 ---
 library_name: transformers
-language:
-- en
 base_model: gokulsrinivasagan/bert_base_lda_5_v1
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - matthews_correlation
 - accuracy
 model-index:
 - name: bert_base_lda_5_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_5_v1_cola
 
-This model is a fine-tuned version of [gokulsrinivasagan/bert_base_lda_5_v1](https://huggingface.co/gokulsrinivasagan/bert_base_lda_5_v1) on the GLUE COLA dataset.
+This model is a fine-tuned version of [gokulsrinivasagan/bert_base_lda_5_v1](https://huggingface.co/gokulsrinivasagan/bert_base_lda_5_v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6183
-- Matthews Correlation: 0.0
-- Accuracy: 0.6913
+- Loss: 0.9654
+- Matthews Correlation: 0.1201
+- Accuracy: 0.6424
 
 ## 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,15 +51,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Matthews Correlation | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------------------:|:--------:|
-| 1.0659        | 1.0   | 34   | 0.6201          | 0.0                  | 0.6913   |
-| 0.6315        | 2.0   | 68   | 0.6200          | 0.0                  | 0.6913   |
-| 0.61          | 3.0   | 102  | 0.6210          | 0.0                  | 0.6913   |
-| 0.6103        | 4.0   | 136  | 0.6183          | 0.0                  | 0.6913   |
-| 0.6103        | 5.0   | 170  | 0.6222          | 0.0                  | 0.6913   |
-| 0.6105        | 6.0   | 204  | 0.6192          | 0.0                  | 0.6913   |
-| 0.6098        | 7.0   | 238  | 0.6203          | 0.0                  | 0.6913   |
-| 0.6104        | 8.0   | 272  | 0.6233          | 0.0                  | 0.6913   |
-| 0.6095        | 9.0   | 306  | 0.6257          | 0.0                  | 0.6913   |
+| 0.6165        | 1.0   | 34   | 0.6256          | 0.0                  | 0.6913   |
+| 0.6085        | 2.0   | 68   | 0.6207          | 0.0438               | 0.6922   |
+| 0.5767        | 3.0   | 102  | 0.6141          | 0.0143               | 0.6788   |
+| 0.5249        | 4.0   | 136  | 0.6586          | 0.0600               | 0.6625   |
+| 0.4572        | 5.0   | 170  | 0.7236          | 0.1336               | 0.6520   |
+| 0.3844        | 6.0   | 204  | 0.8147          | 0.0818               | 0.6405   |
+| 0.327         | 7.0   | 238  | 0.9068          | 0.0894               | 0.6127   |
+| 0.2756        | 8.0   | 272  | 0.9654          | 0.1201               | 0.6424   |
 
 
 ### Framework versions

logs/events.out.tfevents.1733309138.ki-g0008.1206925.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:16d83a407c5c27a93f5eea5d65e6c89f0e784f706a625e94245e9cf5440de62b
-size 5666
+oid sha256:f594701a9c4c8134e757cbb3008a0a0630bfa2f1bc306e5baf520d303f94dc56
+size 10184

model.safetensors

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