Update README.md

README.md

@@ -1,11 +1,11 @@
 ---
-license: mit
-tags:
-- generated_from_trainer
 datasets:
-- imdb
+- seamew/ChnSentiCorp
 metrics:
 - accuracy
+- precision
+- f1
+- recall
 model-index:
 - name: gpt2-imdb-sentiment-classifier
   results:
@@ -20,6 +20,9 @@ model-index:
     - name: Accuracy
       type: accuracy
       value: 0.9394
+language:
+- zh
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -27,14 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # gpt2-imdb-sentiment-classifier
 
-This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the imdb dataset.
+This model is a fine-tuned version of [hfl/rbt6](https://huggingface.co/hfl/rbt6) on the ChnSentiCorp dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1703
-- Accuracy: 0.9394
-
-## Model description
-
-More information needed
+- Loss: 0.294600	
+- Accuracy: 0.933884
 
 ## Intended uses & limitations
 
@@ -51,24 +50,19 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5e-05
-- train_batch_size: 16
-- eval_batch_size: 16
-- seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- num_epochs: 1
-
+- learning_rate: 2e-05
+- train_batch_size: 4
+- eval_batch_size: 4
+- weight_decay=1e-2
+- num_train_epochs=3
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.1967        | 1.0   | 1563 | 0.1703          | 0.9394   |
-
+Epoch	Training Loss	Validation Loss	Accuracy	F1	Precision	Recall
+1	0.359700	0.306089	0.924242	0.926230	0.918699	0.933884
+2	0.200600	0.295512	0.942761	0.943615	0.946755	0.940496
+3	0.105600	0.294600	0.941919	0.942452	0.951178	0.933884
 
 ### Framework versions
 
-- Transformers 4.18.0
-- Pytorch 1.13.1+cu117
-- Datasets 2.9.0
-- Tokenizers 0.12.1
+- Pytorch 2.0.0
+- Python 3.9.12