Model save

README.md

@@ -1,28 +1,13 @@
 ---
 library_name: transformers
-language:
-- en
 base_model: gokulsrinivasagan/bert_tiny_lda_20_v1
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - accuracy
 model-index:
 - name: bert_tiny_lda_20_v1_mnli
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: GLUE MNLI
-      type: glue
-      args: mnli
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.3522172497965826
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert_tiny_lda_20_v1_mnli
 
-This model is a fine-tuned version of [gokulsrinivasagan/bert_tiny_lda_20_v1](https://huggingface.co/gokulsrinivasagan/bert_tiny_lda_20_v1) on the GLUE MNLI dataset.
+This model is a fine-tuned version of [gokulsrinivasagan/bert_tiny_lda_20_v1](https://huggingface.co/gokulsrinivasagan/bert_tiny_lda_20_v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0962
-- Accuracy: 0.3522
+- Loss: 0.8428
+- Accuracy: 0.6963
 
 ## Model description
 
@@ -52,7 +37,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
@@ -64,13 +49,15 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 1.0993        | 1.0   | 1534  | 1.0993          | 0.3274   |
-| 1.0986        | 2.0   | 3068  | 1.0962          | 0.3545   |
-| 1.0987        | 3.0   | 4602  | 1.0964          | 0.3274   |
-| 1.0985        | 4.0   | 6136  | 1.1016          | 0.3182   |
-| 1.0985        | 5.0   | 7670  | 1.0989          | 0.3545   |
-| 1.0986        | 6.0   | 9204  | 1.0989          | 0.3545   |
-| 1.0986        | 7.0   | 10738 | 1.0993          | 0.3182   |
+| 0.9696        | 1.0   | 1534  | 0.8635          | 0.6102   |
+| 0.8307        | 2.0   | 3068  | 0.7849          | 0.6501   |
+| 0.7523        | 3.0   | 4602  | 0.7467          | 0.6728   |
+| 0.6962        | 4.0   | 6136  | 0.7247          | 0.6862   |
+| 0.6472        | 5.0   | 7670  | 0.7248          | 0.6957   |
+| 0.6032        | 6.0   | 9204  | 0.7455          | 0.6984   |
+| 0.5606        | 7.0   | 10738 | 0.7510          | 0.6987   |
+| 0.5204        | 8.0   | 12272 | 0.7849          | 0.6915   |
+| 0.4808        | 9.0   | 13806 | 0.8428          | 0.6963   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:93100b3188b7c55af0a39b44798c0e9283661d429349a20d138424df7cea48a9
+oid sha256:2485a30c32c166b8759eade10f3305793c5e0391d2890fdfd05233925977c96c
 size 131858796