End of training

README.md

@@ -1,7 +1,7 @@
 ---
-base_model: gpt2
+base_model: google-bert/bert-base-uncased
 library_name: peft
-license: mit
+license: apache-2.0
 metrics:
 - accuracy
 - f1
@@ -10,22 +10,22 @@ metrics:
 tags:
 - generated_from_trainer
 model-index:
-- name: gpt2-sst2-sentiment-classifier-lora
+- name: bert-sst2-sentiment-lora
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# gpt2-sst2-sentiment-classifier-lora
+# bert-sst2-sentiment-lora
 
-This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
+This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2636
-- Accuracy: 0.9083
-- F1: 0.9111
-- Precision: 0.8991
-- Recall: 0.9234
+- Loss: 0.2664
+- Accuracy: 0.9094
+- F1: 0.9123
+- Precision: 0.8993
+- Recall: 0.9257
 
 ## Model description
 
@@ -57,9 +57,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.3138        | 1.0   | 4210  | 0.2550          | 0.9014   | 0.9034 | 0.9013    | 0.9054 |
-| 0.2597        | 2.0   | 8420  | 0.2666          | 0.9014   | 0.9061 | 0.8792    | 0.9347 |
-| 0.2436        | 3.0   | 12630 | 0.2636          | 0.9083   | 0.9111 | 0.8991    | 0.9234 |
+| 0.3043        | 1.0   | 4210  | 0.2483          | 0.9128   | 0.9148 | 0.9107    | 0.9189 |
+| 0.2494        | 2.0   | 8420  | 0.2577          | 0.9083   | 0.9109 | 0.9009    | 0.9212 |
+| 0.1861        | 3.0   | 12630 | 0.2664          | 0.9094   | 0.9123 | 0.8993    | 0.9257 |
 
 
 ### Framework versions

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ada83ab5a8e2475de55e17db4321b8667a75a7074b068065f84802e3de1d8b2a
+oid sha256:3469cac36a9a2586c40aac9daab57adc619ce419aa096b8d6b0a8e359198a993
 size 1785816