Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
---
|
|
|
2 |
tags:
|
3 |
-
-
|
|
|
|
|
4 |
model-index:
|
5 |
- name: KoBigBird-KoBart-News-Summarization
|
6 |
results: []
|
@@ -11,25 +14,36 @@ should probably proofread and complete it, then remove this comment. -->
|
|
11 |
|
12 |
# KoBigBird-KoBart-News-Summarization
|
13 |
|
14 |
-
This model is a fine-tuned version of [noahkim/KoBigBird-KoBart-News-Summarization](https://huggingface.co/noahkim/KoBigBird-KoBart-News-Summarization) on
|
15 |
-
|
16 |
-
- Loss: 4.1236
|
17 |
|
18 |
## Model description
|
19 |
|
20 |
-
|
|
|
|
|
|
|
21 |
|
22 |
-
|
23 |
|
24 |
-
|
|
|
|
|
25 |
|
26 |
-
|
|
|
|
|
|
|
27 |
|
28 |
-
|
|
|
|
|
|
|
29 |
|
30 |
-
|
|
|
|
|
31 |
|
32 |
-
### Training hyperparameters
|
33 |
|
34 |
The following hyperparameters were used during training:
|
35 |
- learning_rate: 2e-05
|
|
|
1 |
---
|
2 |
+
language: ko
|
3 |
tags:
|
4 |
+
- summarization
|
5 |
+
- news
|
6 |
+
inference: false
|
7 |
model-index:
|
8 |
- name: KoBigBird-KoBart-News-Summarization
|
9 |
results: []
|
|
|
14 |
|
15 |
# KoBigBird-KoBart-News-Summarization
|
16 |
|
17 |
+
This model is a fine-tuned version of [noahkim/KoBigBird-KoBart-News-Summarization](https://huggingface.co/noahkim/KoBigBird-KoBart-News-Summarization) on the [daekeun-ml/naver-news-summarization-ko](https://huggingface.co/datasets/daekeun-ml/naver-news-summarization-ko)
|
18 |
+
|
|
|
19 |
|
20 |
## Model description
|
21 |
|
22 |
+
<<20221110 Commit>>
|
23 |
+
|
24 |
+
<<KoBigBird-KoBart-News-Summarization ๋ชจ๋ธ ์ค๋ช
>>
|
25 |
+
๋ค์ค๋ฌธ์์์ฝ(Multi-Document-Summarization) Task๋ฅผ ์ํด์ KoBigBird ๋ชจ๋ธ์ Encoder-Decoder๋ชจ๋ธ์ ๋ง๋ค์ด์ ํ์ต์ ์งํํ์ต๋๋ค. KoBigBird๋ฅผ Decoder๋ก ์ฐ๋ ค๊ณ ํ์ผ๋ ์ค๋ฅ๊ฐ ์๊ฒจ์ ์์ฝ์ ํนํ๋ KoBART์ Decoder๋ฅผ ํ์ฉํด์ ๋ชจ๋ธ์ ์์ฑํ์ต๋๋ค.
|
26 |
|
27 |
+
ํ๋ก์ ํธ์ฉ์ผ๋ก ๋ด์ค ์์ฝ ๋ชจ๋ธ ํนํ๋ ๋ชจ๋ธ์ ๋ง๋ค๊ธฐ ์ํด ๊ธฐ์กด์ ๋ง๋ค์๋ KoBigBird-KoBart-News-Summarization ๋ชจ๋ธ์ ์ถ๊ฐ์ ์ผ๋ก daekeun-ml๋์ด ์ ๊ณตํด์ฃผ์ naver-news-summarization-ko ๋ฐ์ดํฐ์
์ผ๋ก ํ์ธํ๋ ํ์ต๋๋ค.
|
28 |
|
29 |
+
ํ์ฌ AI-HUB์์ ์ ๊ณตํ๋ ์์ฝ ๋ฐ์ดํฐ๋ฅผ ์ถ๊ฐ ํ์ต ์งํ ์์ ์
๋๋ค.
|
30 |
+
์ง์์ ์ผ๋ก ๋ฐ์ ์์ผ ์ข์ ์ฑ๋ฅ์ ๋ชจ๋ธ์ ๊ตฌํํ๊ฒ ์ต๋๋ค.
|
31 |
+
๊ฐ์ฌํฉ๋๋ค.
|
32 |
|
33 |
+
์คํํ๊ฒฝ
|
34 |
+
- Google Colab Pro
|
35 |
+
- CPU : Intel(R) Xeon(R) CPU @ 2.20GHz
|
36 |
+
- GPU : A100-SXM4-40GB
|
37 |
|
38 |
+
<pre><code>
|
39 |
+
# Python Code
|
40 |
+
from transformers import AutoTokenizer
|
41 |
+
from transformers import AutoModelForSeq2SeqLM
|
42 |
|
43 |
+
tokenizer = AutoTokenizer.from_pretrained("noahkim/KoT5_news_summarization")
|
44 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("noahkim/KoT5_news_summarization")
|
45 |
+
</pre></code>
|
46 |
|
|
|
47 |
|
48 |
The following hyperparameters were used during training:
|
49 |
- learning_rate: 2e-05
|