Model save
Browse files
    	
        README.md
    ADDED
    
    | @@ -0,0 +1,104 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            ---
         | 
| 2 | 
            +
            license: apache-2.0
         | 
| 3 | 
            +
            base_model: bert-base-uncased
         | 
| 4 | 
            +
            tags:
         | 
| 5 | 
            +
            - generated_from_trainer
         | 
| 6 | 
            +
            datasets:
         | 
| 7 | 
            +
            - emotion
         | 
| 8 | 
            +
            metrics:
         | 
| 9 | 
            +
            - accuracy
         | 
| 10 | 
            +
            model-index:
         | 
| 11 | 
            +
            - name: BERT_Emotions_tuned
         | 
| 12 | 
            +
              results:
         | 
| 13 | 
            +
              - task:
         | 
| 14 | 
            +
                  name: Text Classification
         | 
| 15 | 
            +
                  type: text-classification
         | 
| 16 | 
            +
                dataset:
         | 
| 17 | 
            +
                  name: emotion
         | 
| 18 | 
            +
                  type: emotion
         | 
| 19 | 
            +
                  config: split
         | 
| 20 | 
            +
                  split: validation
         | 
| 21 | 
            +
                  args: split
         | 
| 22 | 
            +
                metrics:
         | 
| 23 | 
            +
                - name: Accuracy
         | 
| 24 | 
            +
                  type: accuracy
         | 
| 25 | 
            +
                  value: 0.9295
         | 
| 26 | 
            +
            ---
         | 
| 27 | 
            +
             | 
| 28 | 
            +
            <!-- This model card has been generated automatically according to the information the Trainer had access to. You
         | 
| 29 | 
            +
            should probably proofread and complete it, then remove this comment. -->
         | 
| 30 | 
            +
             | 
| 31 | 
            +
            # BERT_Emotions_tuned
         | 
| 32 | 
            +
             | 
| 33 | 
            +
            This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the emotion dataset.
         | 
| 34 | 
            +
            It achieves the following results on the evaluation set:
         | 
| 35 | 
            +
            - Loss: 0.2033
         | 
| 36 | 
            +
            - Accuracy: 0.9295
         | 
| 37 | 
            +
             | 
| 38 | 
            +
            ## Model description
         | 
| 39 | 
            +
             | 
| 40 | 
            +
            More information needed
         | 
| 41 | 
            +
             | 
| 42 | 
            +
            ## Intended uses & limitations
         | 
| 43 | 
            +
             | 
| 44 | 
            +
            More information needed
         | 
| 45 | 
            +
             | 
| 46 | 
            +
            ## Training and evaluation data
         | 
| 47 | 
            +
             | 
| 48 | 
            +
            More information needed
         | 
| 49 | 
            +
             | 
| 50 | 
            +
            ## Training procedure
         | 
| 51 | 
            +
             | 
| 52 | 
            +
            ### Training hyperparameters
         | 
| 53 | 
            +
             | 
| 54 | 
            +
            The following hyperparameters were used during training:
         | 
| 55 | 
            +
            - learning_rate: 5e-05
         | 
| 56 | 
            +
            - train_batch_size: 16
         | 
| 57 | 
            +
            - eval_batch_size: 16
         | 
| 58 | 
            +
            - seed: 42
         | 
| 59 | 
            +
            - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
         | 
| 60 | 
            +
            - lr_scheduler_type: linear
         | 
| 61 | 
            +
            - num_epochs: 3
         | 
| 62 | 
            +
             | 
| 63 | 
            +
            ### Training results
         | 
| 64 | 
            +
             | 
| 65 | 
            +
            | Training Loss | Epoch | Step | Validation Loss | Accuracy |
         | 
| 66 | 
            +
            |:-------------:|:-----:|:----:|:---------------:|:--------:|
         | 
| 67 | 
            +
            | No log        | 0.1   | 100  | 0.8098          | 0.7195   |
         | 
| 68 | 
            +
            | No log        | 0.2   | 200  | 0.4054          | 0.882    |
         | 
| 69 | 
            +
            | No log        | 0.3   | 300  | 0.4686          | 0.877    |
         | 
| 70 | 
            +
            | No log        | 0.4   | 400  | 0.2850          | 0.909    |
         | 
| 71 | 
            +
            | 0.5652        | 0.5   | 500  | 0.2673          | 0.92     |
         | 
| 72 | 
            +
            | 0.5652        | 0.6   | 600  | 0.2474          | 0.9255   |
         | 
| 73 | 
            +
            | 0.5652        | 0.7   | 700  | 0.1943          | 0.933    |
         | 
| 74 | 
            +
            | 0.5652        | 0.8   | 800  | 0.1779          | 0.9315   |
         | 
| 75 | 
            +
            | 0.5652        | 0.9   | 900  | 0.1720          | 0.939    |
         | 
| 76 | 
            +
            | 0.2212        | 1.0   | 1000 | 0.1747          | 0.9375   |
         | 
| 77 | 
            +
            | 0.2212        | 1.1   | 1100 | 0.1902          | 0.933    |
         | 
| 78 | 
            +
            | 0.2212        | 1.2   | 1200 | 0.1540          | 0.941    |
         | 
| 79 | 
            +
            | 0.2212        | 1.3   | 1300 | 0.1599          | 0.937    |
         | 
| 80 | 
            +
            | 0.2212        | 1.4   | 1400 | 0.1533          | 0.944    |
         | 
| 81 | 
            +
            | 0.1315        | 1.5   | 1500 | 0.1421          | 0.937    |
         | 
| 82 | 
            +
            | 0.1315        | 1.6   | 1600 | 0.1549          | 0.941    |
         | 
| 83 | 
            +
            | 0.1315        | 1.7   | 1700 | 0.1284          | 0.9435   |
         | 
| 84 | 
            +
            | 0.1315        | 1.8   | 1800 | 0.1376          | 0.934    |
         | 
| 85 | 
            +
            | 0.1315        | 1.9   | 1900 | 0.1197          | 0.943    |
         | 
| 86 | 
            +
            | 0.1204        | 2.0   | 2000 | 0.1319          | 0.9385   |
         | 
| 87 | 
            +
            | 0.1204        | 2.1   | 2100 | 0.1535          | 0.935    |
         | 
| 88 | 
            +
            | 0.1204        | 2.2   | 2200 | 0.1488          | 0.943    |
         | 
| 89 | 
            +
            | 0.1204        | 2.3   | 2300 | 0.1583          | 0.94     |
         | 
| 90 | 
            +
            | 0.1204        | 2.4   | 2400 | 0.1426          | 0.9425   |
         | 
| 91 | 
            +
            | 0.0913        | 2.5   | 2500 | 0.1554          | 0.9395   |
         | 
| 92 | 
            +
            | 0.0913        | 2.6   | 2600 | 0.1458          | 0.944    |
         | 
| 93 | 
            +
            | 0.0913        | 2.7   | 2700 | 0.1504          | 0.943    |
         | 
| 94 | 
            +
            | 0.0913        | 2.8   | 2800 | 0.1621          | 0.9465   |
         | 
| 95 | 
            +
            | 0.0913        | 2.9   | 2900 | 0.1521          | 0.944    |
         | 
| 96 | 
            +
            | 0.0842        | 3.0   | 3000 | 0.1533          | 0.944    |
         | 
| 97 | 
            +
             | 
| 98 | 
            +
             | 
| 99 | 
            +
            ### Framework versions
         | 
| 100 | 
            +
             | 
| 101 | 
            +
            - Transformers 4.38.2
         | 
| 102 | 
            +
            - Pytorch 2.1.0+cu121
         | 
| 103 | 
            +
            - Datasets 2.18.0
         | 
| 104 | 
            +
            - Tokenizers 0.15.2
         | 
    	
        model.safetensors
    CHANGED
    
    | @@ -1,3 +1,3 @@ | |
| 1 | 
             
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            -
            oid sha256: | 
| 3 | 
             
            size 437970952
         | 
|  | |
| 1 | 
             
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:aed43988fe4339520a32c522473e0453f3da5d44bb0525696610fc55e3262e46
         | 
| 3 | 
             
            size 437970952
         | 
    	
        runs/Mar13_06-49-16_cfd6c19a3965/events.out.tfevents.1710312689.cfd6c19a3965.477.0
    CHANGED
    
    | @@ -1,3 +1,3 @@ | |
| 1 | 
             
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            -
            oid sha256: | 
| 3 | 
            -
            size  | 
|  | |
| 1 | 
             
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:967ea9367dfe9f8f579ee1a7e3d977d1d464d1e587f725b544c98f32971d2f1c
         | 
| 3 | 
            +
            size 16969
         | 
    	
        runs/Mar13_06-49-16_cfd6c19a3965/events.out.tfevents.1710314719.cfd6c19a3965.477.1
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:6ca19c26349365c2b7d34d29cba0c56d3765a634e9712b3f70746413e071e58a
         | 
| 3 | 
            +
            size 1057
         |