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)
|