TransferGraph
/

marcelcastrobr_sagemaker-distilbert-emotion-finetuned-lora-ag_news

Text Classification

Model card Files Files and versions Community

hilco commited on Feb 28

Commit

8984d36

•

1 Parent(s): dbbec5c

Finished training.

Files changed (2) hide show

README.md +7 -7
adapter_model.safetensors +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ model-index:
       args: default
     metrics:
     - type: accuracy
-      value: 0.9386842105263158
       name: accuracy
 ---
@@ -34,7 +34,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [marcelcastrobr/sagemaker-distilbert-emotion](https://huggingface.co/marcelcastrobr/sagemaker-distilbert-emotion) on the ag_news dataset.
 It achieves the following results on the evaluation set:
-- accuracy: 0.9387
 ## Model description
@@ -65,11 +65,11 @@ The following hyperparameters were used during training:
 | accuracy | train_loss | epoch |
 |:--------:|:----------:|:-----:|
-| 0.2486   | None       | 0     |
-| 0.9236   | 0.2643     | 0     |
-| 0.9316   | 0.2018     | 1     |
-| 0.9371   | 0.1795     | 2     |
-| 0.9387   | 0.1629     | 3     |
 ### Framework versions

       args: default
     metrics:
     - type: accuracy
+      value: 0.9397368421052632
       name: accuracy
 ---
 This model is a fine-tuned version of [marcelcastrobr/sagemaker-distilbert-emotion](https://huggingface.co/marcelcastrobr/sagemaker-distilbert-emotion) on the ag_news dataset.
 It achieves the following results on the evaluation set:
+- accuracy: 0.9397
 ## Model description
 | accuracy | train_loss | epoch |
 |:--------:|:----------:|:-----:|
+| 0.2130   | None       | 0     |
+| 0.9246   | 0.2626     | 0     |
+| 0.93     | 0.2000     | 1     |
+| 0.9366   | 0.1770     | 2     |
+| 0.9397   | 0.1618     | 3     |
 ### Framework versions

adapter_model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2895b4ba23bb56f280108b774bc078cb5a8ea3dbd1394cfde1c754c1e9c0e1c0
 size 2452208

 version https://git-lfs.github.com/spec/v1
+oid sha256:ab936b2f5c2f2e608f9e4ed596e310d511a9f1b7df7317da609a66d918076a16
 size 2452208