Hugging Face's logo

aedupuga
/
fiction_predictor

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Community
aedupuga commited on
Commit
59f5c41
·
verified ·
1 Parent(s): 6324d29

End of training

Browse files
Files changed (2) hide show
  1. README.md +10 -8
  2. model.safetensors +1 -1
README.md CHANGED
@@ -29,13 +29,15 @@ It achieves the following results on the evaluation set:
29
 
30
  ## Model description
31
 
32
- This model uses text dataset of reviews and classifies them as fiction or not. This model was developed with distilbert-base-uncased.
33
- The data for this model came from jennifee/HW1-image-dataset
34
 
35
  ## Intended uses & limitations
36
 
37
- This model was developed as a practice in fine tuning models trained on image datasets.
38
 
 
 
 
39
 
40
  ## Training procedure
41
 
@@ -54,11 +56,11 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
57
- | 0.0048 | 1.0 | 128 | 0.0202 | 0.9961 | 0.9961 | 0.9961 | 0.9961 |
58
- | 0.0015 | 2.0 | 256 | 0.0025 | 1.0 | 1.0 | 1.0 | 1.0 |
59
- | 0.0009 | 3.0 | 384 | 0.0007 | 1.0 | 1.0 | 1.0 | 1.0 |
60
- | 0.0006 | 4.0 | 512 | 0.0005 | 1.0 | 1.0 | 1.0 | 1.0 |
61
- | 0.0006 | 5.0 | 640 | 0.0004 | 1.0 | 1.0 | 1.0 | 1.0 |
62
 
63
 
64
  ### Framework versions
 
29
 
30
  ## Model description
31
 
32
+ More information needed
 
33
 
34
  ## Intended uses & limitations
35
 
36
+ More information needed
37
 
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
 
42
  ## Training procedure
43
 
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
+ | 0.0045 | 1.0 | 128 | 0.0228 | 0.9922 | 0.9922 | 0.9923 | 0.9922 |
60
+ | 0.0017 | 2.0 | 256 | 0.0012 | 1.0 | 1.0 | 1.0 | 1.0 |
61
+ | 0.001 | 3.0 | 384 | 0.0007 | 1.0 | 1.0 | 1.0 | 1.0 |
62
+ | 0.0007 | 4.0 | 512 | 0.0005 | 1.0 | 1.0 | 1.0 | 1.0 |
63
+ | 0.0006 | 5.0 | 640 | 0.0005 | 1.0 | 1.0 | 1.0 | 1.0 |
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a75e1bebcf3b6710b4c96a0023249f88308bb6a368da163e2ad4dbb4531c0a36
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca607e4ab586e45c380754874c1717c3f72110c9b37ff64f173e6d2335417e7
3
  size 267832560