End of training

README.md

@@ -1,5 +1,9 @@
 ---
-license: mit
+library_name: transformers
+license: apache-2.0
+base_model: distilbert-base-uncased
+tags:
+- generated_from_trainer
 metrics:
 - accuracy
 - f1
@@ -8,8 +12,6 @@ metrics:
 model-index:
 - name: 2025-24679-text-distilbert-predictor
   results: []
-datasets:
-- jennifee/HW1-aug-text-dataset
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -19,7 +21,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0011
+- Loss: 0.0010
 - Accuracy: 1.0
 - F1: 1.0
 - Precision: 1.0
@@ -27,14 +29,18 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-This model predicts if a book is fiction or nonfiction based on book reviews. It is trained with Autogluon multimodal on the jennifee/HW1-aug-text-dataset dataset.
+More information needed
 
 ## Intended uses & limitations
 
-This model is intended to be used to predict if a book is nonfiction or fiction based on a review of the book.
+More information needed
 
 ## Training and evaluation data
 
+More information needed
+
+## Training procedure
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -50,11 +56,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.004         | 1.0   | 128  | 0.0212          | 0.9961   | 0.9961 | 0.9961    | 0.9961 |
-| 0.0017        | 2.0   | 256  | 0.0022          | 1.0      | 1.0    | 1.0       | 1.0    |
+| 0.0042        | 1.0   | 128  | 0.0289          | 0.9922   | 0.9922 | 0.9923    | 0.9922 |
+| 0.0018        | 2.0   | 256  | 0.0017          | 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    |
+| 0.0006        | 5.0   | 640  | 0.0005          | 1.0      | 1.0    | 1.0       | 1.0    |
 
 
 ### Framework versions
@@ -62,4 +68,4 @@ The following hyperparameters were used during training:
 - Transformers 4.56.1
 - Pytorch 2.8.0+cu126
 - Datasets 4.0.0
-- Tokenizers 0.22.0
+- Tokenizers 0.22.0

model.safetensors

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