Yuchan commited on
Update AlphaS2S.py
Browse files- AlphaS2S.py +2 -2
AlphaS2S.py
CHANGED
|
@@ -296,7 +296,7 @@ chat_model.save_weights("chat_model.weights.h5")
|
|
| 296 |
print("\nโ
๋ชจ๋ธ ๊ฐ์ค์น ์ ์ฅ ์๋ฃ!")
|
| 297 |
|
| 298 |
|
| 299 |
-
def generate_translation_beam(model, input_text, max_len=
|
| 300 |
# Encoder input
|
| 301 |
enc_ids = text_to_ids(input_text)
|
| 302 |
enc_ids = enc_ids[-max_len:]
|
|
@@ -342,5 +342,5 @@ def generate_translation_beam(model, input_text, max_len=220, beam_width=5):
|
|
| 342 |
|
| 343 |
# ์ฌ์ฉ ์์
|
| 344 |
src_text = "์๋
ํ์ธ์! ์ค๋ ๋ ์จ๋ ์ด๋์?"
|
| 345 |
-
translation = generate_translation_beam(chat_model, src_text, max_len=
|
| 346 |
print("๋ฒ์ญ ๊ฒฐ๊ณผ:", translation)
|
|
|
|
| 296 |
print("\nโ
๋ชจ๋ธ ๊ฐ์ค์น ์ ์ฅ ์๋ฃ!")
|
| 297 |
|
| 298 |
|
| 299 |
+
def generate_translation_beam(model, input_text, max_len=128, beam_width=5):
|
| 300 |
# Encoder input
|
| 301 |
enc_ids = text_to_ids(input_text)
|
| 302 |
enc_ids = enc_ids[-max_len:]
|
|
|
|
| 342 |
|
| 343 |
# ์ฌ์ฉ ์์
|
| 344 |
src_text = "์๋
ํ์ธ์! ์ค๋ ๋ ์จ๋ ์ด๋์?"
|
| 345 |
+
translation = generate_translation_beam(chat_model, src_text, max_len=128, beam_width=5)
|
| 346 |
print("๋ฒ์ญ ๊ฒฐ๊ณผ:", translation)
|