aedupuga
/

fiction_predictor

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

aedupuga commited on Sep 23, 2025

Commit

59f5c41

·

verified ·

1 Parent(s): 6324d29

End of training

Files changed (2) hide show

README.md +10 -8
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -29,13 +29,15 @@ It achieves the following results on the evaluation set:
 ## Model description
-This model uses text dataset of reviews and classifies them as fiction or not. This model was developed with distilbert-base-uncased.
-The data for this model came from jennifee/HW1-image-dataset
 ## Intended uses & limitations
-This model was developed as a practice in fine tuning models trained on image datasets.
 ## Training procedure
@@ -54,11 +56,11 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.0048        | 1.0   | 128  | 0.0202          | 0.9961   | 0.9961 | 0.9961    | 0.9961 |
-| 0.0015        | 2.0   | 256  | 0.0025          | 1.0      | 1.0    | 1.0       | 1.0    |
-| 0.0009        | 3.0   | 384  | 0.0007          | 1.0      | 1.0    | 1.0       | 1.0    |
-| 0.0006        | 4.0   | 512  | 0.0005          | 1.0      | 1.0    | 1.0       | 1.0    |
-| 0.0006        | 5.0   | 640  | 0.0004          | 1.0      | 1.0    | 1.0       | 1.0    |
 ### Framework versions

 ## Model description
+More information needed
 ## Intended uses & limitations
+More information needed
+## Training and evaluation data
+More information needed
 ## Training procedure
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| 0.0045        | 1.0   | 128  | 0.0228          | 0.9922   | 0.9922 | 0.9923    | 0.9922 |
+| 0.0017        | 2.0   | 256  | 0.0012          | 1.0      | 1.0    | 1.0       | 1.0    |
+| 0.001         | 3.0   | 384  | 0.0007          | 1.0      | 1.0    | 1.0       | 1.0    |
+| 0.0007        | 4.0   | 512  | 0.0005          | 1.0      | 1.0    | 1.0       | 1.0    |
+| 0.0006        | 5.0   | 640  | 0.0005          | 1.0      | 1.0    | 1.0       | 1.0    |
 ### Framework versions

model.safetensors CHANGED Viewed

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

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