tvaldez
/

classifier-chapter6

Text Classification

Generated from Trainer

Model card Files Files and versions

tvaldez commited on Apr 7

Commit

d27d96d

·

verified ·

1 Parent(s): 5c7d15f

End of training

Files changed (2) hide show

README.md +5 -5
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -19,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2469
-- Accuracy: 0.9207
-- F1: 0.9206
 ## Model description
@@ -52,8 +52,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| No log        | 1.0   | 313  | 0.2637          | 0.9088   | 0.9085 |
-| 0.301         | 2.0   | 626  | 0.2469          | 0.9207   | 0.9206 |
 ### Framework versions

 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.2484
+- Accuracy: 0.9191
+- F1: 0.9191
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| No log        | 1.0   | 313  | 0.2710          | 0.91     | 0.9099 |
+| 0.3018        | 2.0   | 626  | 0.2484          | 0.9191   | 0.9191 |
 ### Framework versions

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9b9906f296ce020d150aa386520e43875a0099ec0084ebee5876c182eb66b1ea
 size 267838720

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