Model save

README.md

@@ -0,0 +1,70 @@
+---
+license: apache-2.0
+base_model: facebook/bart-base
+tags:
+- generated_from_trainer
+metrics:
+- rouge
+model-index:
+- name: BART-finetuned-BBC
+  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. -->
+
+# BART-finetuned-BBC
+
+This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.5563
+- Rouge1: 0.2171
+- Rouge2: 0.1721
+- Rougel: 0.2038
+- Rougelsum: 0.2041
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5.6e-05
+- train_batch_size: 8
+- eval_batch_size: 8
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 8
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
+| 1.7637        | 1.0   | 223  | 2.5862          | 0.212  | 0.1654 | 0.1983 | 0.1987    |
+| 2.1325        | 2.0   | 446  | 2.5126          | 0.2152 | 0.171  | 0.2017 | 0.2017    |
+| 1.9421        | 3.0   | 669  | 2.4889          | 0.2113 | 0.1653 | 0.1973 | 0.1975    |
+| 1.791         | 4.0   | 892  | 2.4865          | 0.2143 | 0.1708 | 0.2019 | 0.2017    |
+| 1.6626        | 5.0   | 1115 | 2.4988          | 0.2148 | 0.1694 | 0.2007 | 0.2006    |
+| 1.5752        | 6.0   | 1338 | 2.5300          | 0.2139 | 0.1692 | 0.2009 | 0.2011    |
+| 1.4931        | 7.0   | 1561 | 2.5403          | 0.2165 | 0.1715 | 0.2028 | 0.2031    |
+| 1.4494        | 8.0   | 1784 | 2.5563          | 0.2171 | 0.1721 | 0.2038 | 0.2041    |
+
+
+### Framework versions
+
+- Transformers 4.35.2
+- Pytorch 2.1.0+cu118
+- Datasets 2.15.0
+- Tokenizers 0.15.0

generation_config.json

@@ -0,0 +1,12 @@
+{
+  "bos_token_id": 0,
+  "decoder_start_token_id": 2,
+  "early_stopping": true,
+  "eos_token_id": 2,
+  "forced_bos_token_id": 0,
+  "forced_eos_token_id": 2,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 4,
+  "pad_token_id": 1,
+  "transformers_version": "4.35.2"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:92eb89bf94caf4e32af3f3c58951c287aee00b95dadcb52716f1c66b8ed021fd
+oid sha256:616f5794bf2ce10ec2c24f8d6a2a8774b4f17aca32240c41bd6691ac445bf017
 size 557912620

runs/Dec13_11-10-26_9bfb7a1b058f/events.out.tfevents.1702465829.9bfb7a1b058f.14417.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1eb621a823b11845b98eb4ea808ca83c552695a4ab89a31c670309c983e9e9c6
-size 9232
+oid sha256:f85b09e81d038a30c4be371ad50a4d0df629e6917e4be4a9f2692e9c666196af
+size 10848