End of training

README.md

@@ -3,6 +3,8 @@ license: mit
 base_model: sanalsprasad/ai-generated-text-classification
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: train_debug
   results: []
@@ -14,6 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 # train_debug
 
 This model is a fine-tuned version of [sanalsprasad/ai-generated-text-classification](https://huggingface.co/sanalsprasad/ai-generated-text-classification) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.6810
+- Accuracy: 0.579
+- Roc Auc: 0.5385
 
 ## Model description
 
@@ -40,9 +46,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 2
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Accuracy | Roc Auc |
+|:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------:|
+| 0.6814        | 1.0   | 6250  | 0.6806          | 0.579    | 0.4785  |
+| 0.6761        | 2.0   | 12500 | 0.6810          | 0.579    | 0.5385  |
+
+
 ### Framework versions
 
 - Transformers 4.36.0
-- Pytorch 2.0.0+cpu
+- Pytorch 2.0.0
 - Datasets 2.1.0
 - Tokenizers 0.15.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fd85f92c7f1b73f0fba0ee7416cfea2e913d5a3d83b01ae3cb5c0d6663d8d4ec
+oid sha256:27e1a06bc91fea08b51b3f6e08ddc7861b26cf760d1d74d8b46f53d072ac4367
 size 498612824

runs/Jan19_20-12-01_ff121b75a6a8/events.out.tfevents.1705695125.ff121b75a6a8.26.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:769c5cd268183c97eea2fa959bb357763399e5bfc2cd6eb070d38118812110f0
+size 9622

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4ee8efd227516884400980651729aebba1c2524d13056e61af3d95b5a7f47491
-size 4219
+oid sha256:4414eeb8d969b11f73e1a1aa522f520c9020807f8201315a2b8ff6b2f0775b21
+size 4283