Model save

README.md

@@ -1,28 +1,13 @@
 ---
 library_name: transformers
-language:
-- en
 base_model: gokulsrinivasagan/distilbert_lda_50_v1
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - accuracy
 model-index:
 - name: distilbert_lda_50_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.3295362082994304
+  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. -->
 
 # distilbert_lda_50_v1_mnli
 
-This model is a fine-tuned version of [gokulsrinivasagan/distilbert_lda_50_v1](https://huggingface.co/gokulsrinivasagan/distilbert_lda_50_v1) on the GLUE MNLI dataset.
+This model is a fine-tuned version of [gokulsrinivasagan/distilbert_lda_50_v1](https://huggingface.co/gokulsrinivasagan/distilbert_lda_50_v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0962
-- Accuracy: 0.3295
+- Loss: 0.9725
+- Accuracy: 0.7280
 
 ## 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,20 +49,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 1.1024        | 1.0   | 1534  | 1.0993          | 0.3274   |
-| 1.0986        | 2.0   | 3068  | 1.0987          | 0.3545   |
-| 1.0987        | 3.0   | 4602  | 1.0989          | 0.3274   |
-| 1.0986        | 4.0   | 6136  | 1.1016          | 0.3182   |
-| 1.0985        | 5.0   | 7670  | 1.0989          | 0.3545   |
-| 1.0987        | 6.0   | 9204  | 1.0989          | 0.3545   |
-| 1.0985        | 7.0   | 10738 | 1.0968          | 0.3182   |
-| 1.0986        | 8.0   | 12272 | 1.0991          | 0.3182   |
-| 1.0988        | 9.0   | 13806 | 1.0962          | 0.3274   |
-| 1.0986        | 10.0  | 15340 | 1.0992          | 0.3274   |
-| 1.0986        | 11.0  | 16874 | 1.0990          | 0.3274   |
-| 1.0984        | 12.0  | 18408 | 1.0991          | 0.3182   |
-| 1.0985        | 13.0  | 19942 | 1.0965          | 0.3182   |
-| 1.0988        | 14.0  | 21476 | 1.0987          | 0.3274   |
+| 0.8292        | 1.0   | 1534  | 0.7243          | 0.6906   |
+| 0.6692        | 2.0   | 3068  | 0.6628          | 0.7185   |
+| 0.5756        | 3.0   | 4602  | 0.6567          | 0.7336   |
+| 0.4939        | 4.0   | 6136  | 0.6776          | 0.7320   |
+| 0.414         | 5.0   | 7670  | 0.6913          | 0.7373   |
+| 0.3422        | 6.0   | 9204  | 0.7809          | 0.7314   |
+| 0.2807        | 7.0   | 10738 | 0.8294          | 0.7350   |
+| 0.2292        | 8.0   | 12272 | 0.9725          | 0.7280   |
 
 
 ### Framework versions

logs/events.out.tfevents.1733313788.ki-g0008.1206436.16

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:95aff9b6b5f8f68f438db6d95e96e04f983f46cda5ad3d775ada66315dfb52ec
-size 8926
+oid sha256:c7908388c9a02754dfbe0015651169f5cbadd6f6c668919169399c7e7e182d32
+size 9814

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:93a42abe8758e640259cda2167237ce9a1d3f767b9d6b62f737f3fef9f361062
+oid sha256:8b6bd7fcb8b43122ad16bc32304adc642424443acd1c7553fd7ff299903b661a
 size 267835644