Amirth24
/

disease_classifier

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Amirth24 commited on Sep 23

Commit

25d7456

•

1 Parent(s): ff61e7e

End of training

Files changed (2) hide show

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

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ 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: 1.3195
-- Accuracy: 0.8254
 ## Model description
@@ -50,8 +50,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 353  | 1.3662          | 0.8183   |
-| 1.4983        | 2.0   | 706  | 1.3195          | 0.8254   |
 ### 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: 1.3954
+- Accuracy: 0.8353
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 1.0   | 353  | 1.3158          | 0.8297   |
+| 0.981         | 2.0   | 706  | 1.3954          | 0.8353   |
 ### Framework versions

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:05069677cd43a567eba1a4c7b2168b4cc16e7188c68c012240ca09af4801dcd3
 size 271154672

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