Yuchan commited on
Commit
094d892
ยท
verified ยท
1 Parent(s): 6c265e3

Update AlphaS2S.py

Browse files
Files changed (1) hide show
  1. 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=220, beam_width=5):
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=220, beam_width=5)
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)