End of training

README.md

@@ -1,199 +1,65 @@
 ---
 library_name: transformers
-tags: []
+license: apache-2.0
+base_model: distilbert/distilbert-base-uncased
+tags:
+- generated_from_trainer
+metrics:
+- accuracy
+- f1
+- precision
+- recall
+model-index:
+- name: DistriBert_v10-1
+  results: []
 ---
 
-# Model Card for Model ID
+<!-- 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. -->
 
-<!-- Provide a quick summary of what the model is/does. -->
+# DistriBert_v10-1
 
+This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the None dataset.
+It achieves the following results on the evaluation set:
+- Accuracy: 0.9135
+- F1: 0.9115
+- Precision: 0.9122
+- Recall: 0.9128
+- Loss: 0.3513
 
+## Model description
 
-## Model Details
+More information needed
 
-### Model Description
+## Intended uses & limitations
 
-<!-- Provide a longer summary of what this model is. -->
+More information needed
 
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
+## Training and evaluation data
 
-- **Developed by:** [More Information Needed]
-- **Funded by [optional]:** [More Information Needed]
-- **Shared by [optional]:** [More Information Needed]
-- **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
-- **Finetuned from model [optional]:** [More Information Needed]
+More information needed
 
-### Model Sources [optional]
+## Training procedure
 
-<!-- Provide the basic links for the model. -->
+### Training hyperparameters
 
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 32
+- eval_batch_size: 32
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine_with_restarts
+- lr_scheduler_warmup_steps: 100
+- num_epochs: 20
 
-## Uses
+### Training results
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-### Direct Use
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+### Framework versions
 
-[More Information Needed]
-
-### Downstream Use [optional]
-
-<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
-
-[More Information Needed]
-
-### Out-of-Scope Use
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
-
-[More Information Needed]
-
-## Bias, Risks, and Limitations
-
-<!-- This section is meant to convey both technical and sociotechnical limitations. -->
-
-[More Information Needed]
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
-## How to Get Started with the Model
-
-Use the code below to get started with the model.
-
-[More Information Needed]
-
-## Training Details
-
-### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-[More Information Needed]
-
-### Training Procedure
-
-<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
-
-#### Preprocessing [optional]
-
-[More Information Needed]
-
-
-#### Training Hyperparameters
-
-- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
-
-#### Speeds, Sizes, Times [optional]
-
-<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
-
-[More Information Needed]
-
-## Evaluation
-
-<!-- This section describes the evaluation protocols and provides the results. -->
-
-### Testing Data, Factors & Metrics
-
-#### Testing Data
-
-<!-- This should link to a Dataset Card if possible. -->
-
-[More Information Needed]
-
-#### Factors
-
-<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
-
-[More Information Needed]
-
-#### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
-
-[More Information Needed]
-
-### Results
-
-[More Information Needed]
-
-#### Summary
-
-
-
-## Model Examination [optional]
-
-<!-- Relevant interpretability work for the model goes here -->
-
-[More Information Needed]
-
-## Environmental Impact
-
-<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
-
-Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
-
-- **Hardware Type:** [More Information Needed]
-- **Hours used:** [More Information Needed]
-- **Cloud Provider:** [More Information Needed]
-- **Compute Region:** [More Information Needed]
-- **Carbon Emitted:** [More Information Needed]
-
-## Technical Specifications [optional]
-
-### Model Architecture and Objective
-
-[More Information Needed]
-
-### Compute Infrastructure
-
-[More Information Needed]
-
-#### Hardware
-
-[More Information Needed]
-
-#### Software
-
-[More Information Needed]
-
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
-
-**BibTeX:**
-
-[More Information Needed]
-
-**APA:**
-
-[More Information Needed]
-
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
-## Model Card Authors [optional]
-
-[More Information Needed]
-
-## Model Card Contact
-
-[More Information Needed]
+- Transformers 4.44.2
+- Pytorch 2.5.0+cu121
+- Datasets 3.1.0
+- Tokenizers 0.19.1

all_results.json

@@ -0,0 +1,21 @@
+{
+    "epoch": 10.0,
+    "eval_accuracy": 0.9134920634920635,
+    "eval_f1": 0.9115252835945801,
+    "eval_loss": 0.3512510657310486,
+    "eval_precision": 0.9122290522908707,
+    "eval_recall": 0.9127652441899627,
+    "eval_runtime": 61.9253,
+    "eval_samples_per_second": 61.041,
+    "eval_steps_per_second": 1.922,
+    "total_flos": 1.16919589957632e+16,
+    "train_eval_accuracy": 0.9672335600907029,
+    "train_eval_f1": 0.9673423681503462,
+    "train_eval_loss": 0.11652734130620956,
+    "train_eval_precision": 0.967673238789642,
+    "train_eval_recall": 0.9673335278419569,
+    "train_loss": 0.7689362014549366,
+    "train_runtime": 6188.3292,
+    "train_samples_per_second": 28.505,
+    "train_steps_per_second": 0.892
+}

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "DistriBert_v10-1",
+  "_name_or_path": "distilbert/distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"

eval_results.json

@@ -0,0 +1,11 @@
+{
+    "epoch": 10.0,
+    "eval_accuracy": 0.9134920634920635,
+    "eval_f1": 0.9115252835945801,
+    "eval_loss": 0.3512510657310486,
+    "eval_precision": 0.9122290522908707,
+    "eval_recall": 0.9127652441899627,
+    "eval_runtime": 61.9253,
+    "eval_samples_per_second": 61.041,
+    "eval_steps_per_second": 1.922
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 10.0,
+    "total_flos": 1.16919589957632e+16,
+    "train_loss": 0.7689362014549366,
+    "train_runtime": 6188.3292,
+    "train_samples_per_second": 28.505,
+    "train_steps_per_second": 0.892
+}

trainer_state.json

@@ -0,0 +1,330 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 2760,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "step": 276,
+      "train_eval_accuracy": 0.4954648526077097,
+      "train_eval_f1": 0.424904604475739,
+      "train_eval_loss": 2.101224660873413,
+      "train_eval_precision": 0.4856527073524001,
+      "train_eval_recall": 0.4926378954436191,
+      "train_loss": 2.101224660873413,
+      "train_runtime": 144.8838,
+      "train_samples_per_second": 60.876,
+      "train_steps_per_second": 1.905
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.4835978835978836,
+      "eval_f1": 0.4135688868128075,
+      "eval_loss": 2.1065914630889893,
+      "eval_precision": 0.4476136879464602,
+      "eval_recall": 0.48953657594776134,
+      "eval_runtime": 62.0697,
+      "eval_samples_per_second": 60.899,
+      "eval_steps_per_second": 1.917,
+      "step": 276
+    },
+    {
+      "epoch": 2.0,
+      "step": 552,
+      "train_eval_accuracy": 0.7447845804988662,
+      "train_eval_f1": 0.7128821278125044,
+      "train_eval_loss": 1.0038776397705078,
+      "train_eval_precision": 0.7489107215212647,
+      "train_eval_recall": 0.7458421207363585,
+      "train_loss": 1.0038776397705078,
+      "train_runtime": 145.0361,
+      "train_samples_per_second": 60.812,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.7428571428571429,
+      "eval_f1": 0.7089773330055067,
+      "eval_loss": 1.0131804943084717,
+      "eval_precision": 0.7421511367537219,
+      "eval_recall": 0.7406592004709303,
+      "eval_runtime": 62.0776,
+      "eval_samples_per_second": 60.892,
+      "eval_steps_per_second": 1.917,
+      "step": 552
+    },
+    {
+      "epoch": 3.0,
+      "step": 828,
+      "train_eval_accuracy": 0.8630385487528345,
+      "train_eval_f1": 0.8603167863727107,
+      "train_eval_loss": 0.5634090304374695,
+      "train_eval_precision": 0.8737026471608341,
+      "train_eval_recall": 0.8625907729727306,
+      "train_loss": 0.5634090304374695,
+      "train_runtime": 144.9979,
+      "train_samples_per_second": 60.828,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8481481481481481,
+      "eval_f1": 0.8456142462826861,
+      "eval_loss": 0.5928145051002502,
+      "eval_precision": 0.8571463898409206,
+      "eval_recall": 0.8489798133243263,
+      "eval_runtime": 62.0895,
+      "eval_samples_per_second": 60.88,
+      "eval_steps_per_second": 1.917,
+      "step": 828
+    },
+    {
+      "epoch": 4.0,
+      "step": 1104,
+      "train_eval_accuracy": 0.9109977324263039,
+      "train_eval_f1": 0.9104882432709309,
+      "train_eval_loss": 0.3656206429004669,
+      "train_eval_precision": 0.9163348905783797,
+      "train_eval_recall": 0.9110465902769607,
+      "train_loss": 0.3656206727027893,
+      "train_runtime": 144.8557,
+      "train_samples_per_second": 60.888,
+      "train_steps_per_second": 1.905
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8962962962962963,
+      "eval_f1": 0.8935631085204128,
+      "eval_loss": 0.4208849370479584,
+      "eval_precision": 0.8990702798318864,
+      "eval_recall": 0.8957702000784453,
+      "eval_runtime": 62.1675,
+      "eval_samples_per_second": 60.803,
+      "eval_steps_per_second": 1.914,
+      "step": 1104
+    },
+    {
+      "epoch": 5.0,
+      "step": 1380,
+      "train_eval_accuracy": 0.9232426303854875,
+      "train_eval_f1": 0.9229682041537378,
+      "train_eval_loss": 0.2874959707260132,
+      "train_eval_precision": 0.9273281861728557,
+      "train_eval_recall": 0.9227952922309719,
+      "train_loss": 0.28749600052833557,
+      "train_runtime": 145.0525,
+      "train_samples_per_second": 60.806,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9029100529100529,
+      "eval_f1": 0.9011946016504275,
+      "eval_loss": 0.37733063101768494,
+      "eval_precision": 0.9049954124488648,
+      "eval_recall": 0.9036354818715393,
+      "eval_runtime": 62.1416,
+      "eval_samples_per_second": 60.829,
+      "eval_steps_per_second": 1.915,
+      "step": 1380
+    },
+    {
+      "epoch": 6.0,
+      "step": 1656,
+      "train_eval_accuracy": 0.9370748299319728,
+      "train_eval_f1": 0.9368912367928741,
+      "train_eval_loss": 0.23322050273418427,
+      "train_eval_precision": 0.939204184223903,
+      "train_eval_recall": 0.937322301823473,
+      "train_loss": 0.23322050273418427,
+      "train_runtime": 144.9845,
+      "train_samples_per_second": 60.834,
+      "train_steps_per_second": 1.904
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9105820105820106,
+      "eval_f1": 0.908396393828423,
+      "eval_loss": 0.3413718342781067,
+      "eval_precision": 0.9102975744779533,
+      "eval_recall": 0.9096675456615884,
+      "eval_runtime": 62.1169,
+      "eval_samples_per_second": 60.853,
+      "eval_steps_per_second": 1.916,
+      "step": 1656
+    },
+    {
+      "epoch": 7.0,
+      "step": 1932,
+      "train_eval_accuracy": 0.9435374149659864,
+      "train_eval_f1": 0.943453224205871,
+      "train_eval_loss": 0.20147815346717834,
+      "train_eval_precision": 0.9469882171165018,
+      "train_eval_recall": 0.9435449695196283,
+      "train_loss": 0.20147816836833954,
+      "train_runtime": 144.7624,
+      "train_samples_per_second": 60.927,
+      "train_steps_per_second": 1.907
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9121693121693122,
+      "eval_f1": 0.9103139018247439,
+      "eval_loss": 0.3457355499267578,
+      "eval_precision": 0.9132402827313028,
+      "eval_recall": 0.9120503817924005,
+      "eval_runtime": 62.1165,
+      "eval_samples_per_second": 60.853,
+      "eval_steps_per_second": 1.916,
+      "step": 1932
+    },
+    {
+      "epoch": 8.0,
+      "step": 2208,
+      "train_eval_accuracy": 0.9518140589569161,
+      "train_eval_f1": 0.9517198736909416,
+      "train_eval_loss": 0.1689794510602951,
+      "train_eval_precision": 0.9547315428415143,
+      "train_eval_recall": 0.9519285064221675,
+      "train_loss": 0.16897942125797272,
+      "train_runtime": 145.0413,
+      "train_samples_per_second": 60.81,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9119047619047619,
+      "eval_f1": 0.9099505788900774,
+      "eval_loss": 0.3478126525878906,
+      "eval_precision": 0.9135423252017797,
+      "eval_recall": 0.9116997470551864,
+      "eval_runtime": 62.1667,
+      "eval_samples_per_second": 60.804,
+      "eval_steps_per_second": 1.914,
+      "step": 2208
+    },
+    {
+      "epoch": 9.0,
+      "step": 2484,
+      "train_eval_accuracy": 0.9597505668934241,
+      "train_eval_f1": 0.9598126383463436,
+      "train_eval_loss": 0.13835597038269043,
+      "train_eval_precision": 0.9614206068146748,
+      "train_eval_recall": 0.9599367994432203,
+      "train_loss": 0.13835598528385162,
+      "train_runtime": 144.9973,
+      "train_samples_per_second": 60.829,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.9132275132275133,
+      "eval_f1": 0.9114074718878633,
+      "eval_loss": 0.3454839885234833,
+      "eval_precision": 0.9133521030992556,
+      "eval_recall": 0.91269076612755,
+      "eval_runtime": 62.1786,
+      "eval_samples_per_second": 60.793,
+      "eval_steps_per_second": 1.914,
+      "step": 2484
+    },
+    {
+      "epoch": 10.0,
+      "step": 2760,
+      "train_eval_accuracy": 0.9672335600907029,
+      "train_eval_f1": 0.9673423681503462,
+      "train_eval_loss": 0.11652734130620956,
+      "train_eval_precision": 0.967673238789642,
+      "train_eval_recall": 0.9673335278419569,
+      "train_loss": 0.11652734875679016,
+      "train_runtime": 145.0683,
+      "train_samples_per_second": 60.799,
+      "train_steps_per_second": 1.903
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.9134920634920635,
+      "eval_f1": 0.9115252835945801,
+      "eval_loss": 0.3512510657310486,
+      "eval_precision": 0.9122290522908707,
+      "eval_recall": 0.9127652441899627,
+      "eval_runtime": 62.1606,
+      "eval_samples_per_second": 60.81,
+      "eval_steps_per_second": 1.914,
+      "step": 2760
+    },
+    {
+      "epoch": 10.0,
+      "step": 2760,
+      "total_flos": 1.16919589957632e+16,
+      "train_loss": 0.7689362014549366,
+      "train_runtime": 6188.3292,
+      "train_samples_per_second": 28.505,
+      "train_steps_per_second": 0.892
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.9134920634920635,
+      "eval_f1": 0.9115252835945801,
+      "eval_loss": 0.3512510657310486,
+      "eval_precision": 0.9122290522908707,
+      "eval_recall": 0.9127652441899627,
+      "eval_runtime": 62.1566,
+      "eval_samples_per_second": 60.814,
+      "eval_steps_per_second": 1.915,
+      "step": 2760
+    },
+    {
+      "epoch": 10.0,
+      "step": 2760,
+      "train_en_eval_accuracy": 0.9672335600907029,
+      "train_en_eval_f1": 0.9673423681503462,
+      "train_en_eval_loss": 0.11652734130620956,
+      "train_en_eval_precision": 0.967673238789642,
+      "train_en_eval_recall": 0.9673335278419569,
+      "train_en_loss": 0.11652734875679016,
+      "train_en_runtime": 145.0476,
+      "train_en_samples_per_second": 60.808,
+      "train_en_steps_per_second": 1.903
+    },
+    {
+      "epoch": 10.0,
+      "step": 2760,
+      "test_en_eval_accuracy": 0.9134920634920635,
+      "test_en_eval_f1": 0.9115252835945801,
+      "test_en_eval_loss": 0.3512510657310486,
+      "test_en_eval_precision": 0.9122290522908707,
+      "test_en_eval_recall": 0.9127652441899627,
+      "test_en_loss": 0.3512510657310486,
+      "test_en_runtime": 62.1457,
+      "test_en_samples_per_second": 60.825,
+      "test_en_steps_per_second": 1.915
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 5520,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 20,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.16919589957632e+16,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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