Update README.md
Browse files
README.md
CHANGED
@@ -6,6 +6,18 @@ tags:
|
|
6 |
- summarization
|
7 |
- pegasus_x
|
8 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
This repository features a fine-tuned Pegasus X model designed for summarizing Thai text. The architecture of the model is based on the Pegasus X model.
|
10 |
|
11 |
# Library
|
@@ -28,4 +40,24 @@ summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
|
28 |
|
29 |
print("Input:", new_input_string)
|
30 |
print("Generated Summary:", summary)
|
31 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
- summarization
|
7 |
- pegasus_x
|
8 |
---
|
9 |
+
widget:
|
10 |
+
- text: >-
|
11 |
+
ผมกินตับหมูดิบแล้วหมดสติไป พอฟื้นอีกทีในต่างโลกดันกลายเป็นหมูซะงั้น!
|
12 |
+
คนที่ช่วยผมเอาไว้คือเจส สาวน้อยผู้อ่านใจคนได้ อู๊ด! น่ารัก!
|
13 |
+
ระดับสายตาหมูทำให้เห็นอะไรสีขาวบริสุทธิ์แวบๆ แจ่มเลย... “เอ่อ
|
14 |
+
ฉันได้ยินเสียงในใจของคุณนะคะ…” ฉิบแล้ว! ความมักมากรั่วไหลหมด!
|
15 |
+
แม้ว่าชีวิตประสาหมูที่มีเด็กสาวผู้อ่อนโยนคอยดูแลจะไม่เลว
|
16 |
+
ผมก็ตัดสินใจมุ่งหน้าสู่นครหลวงพร้อมกับเจสเพื่อหาทางกลับเป็นมนุษย์
|
17 |
+
การเดินทางแสนรื่นรมย์จึงเริ่มต้นขึ้น...
|
18 |
+
แต่ไหงเราถึงถูกตามล่าเอาชีวิตล่ะเนี่ย!?
|
19 |
+
example_title: Novel
|
20 |
+
|
21 |
This repository features a fine-tuned Pegasus X model designed for summarizing Thai text. The architecture of the model is based on the Pegasus X model.
|
22 |
|
23 |
# Library
|
|
|
40 |
|
41 |
print("Input:", new_input_string)
|
42 |
print("Generated Summary:", summary)
|
43 |
+
```
|
44 |
+
|
45 |
+
# Training hyperparameters
|
46 |
+
The following hyperparameters were used during training:
|
47 |
+
<ul>
|
48 |
+
<li>accumulation_steps:2</li>
|
49 |
+
<li>num_epochs:20</li>
|
50 |
+
<li>num_beams:6</li>
|
51 |
+
<li>learning_rate:lr=5e-5</li>
|
52 |
+
<li>activation_function:gelu</li>
|
53 |
+
<li>add_bias_logits:True</li>
|
54 |
+
<li>normalize_embedding:True</li>
|
55 |
+
<li>add_final_layer_norm:False </li>
|
56 |
+
<li>normalize_before:False</li>
|
57 |
+
</ul>
|
58 |
+
|
59 |
+
|
60 |
+
|
61 |
+
|
62 |
+
|
63 |
+
|