Commit
·
0804813
1
Parent(s):
c7c6006
Update main_dl.py
Browse files- main_dl.py +2 -2
main_dl.py
CHANGED
@@ -37,7 +37,7 @@ def load_vocab(file_path):
|
|
37 |
return file.read().split('\n')[:-1]
|
38 |
|
39 |
|
40 |
-
|
41 |
global translation_model
|
42 |
|
43 |
vocab_size = 15000
|
@@ -73,7 +73,7 @@ async def decode_sequence_rnn(input_sentence, src, tgt):
|
|
73 |
decoded_sentence += " " + sampled_token
|
74 |
if sampled_token == "[end]":
|
75 |
break
|
76 |
-
return
|
77 |
|
78 |
# ===== Enf of Keras ====
|
79 |
|
|
|
37 |
return file.read().split('\n')[:-1]
|
38 |
|
39 |
|
40 |
+
def decode_sequence_rnn(input_sentence, src, tgt):
|
41 |
global translation_model
|
42 |
|
43 |
vocab_size = 15000
|
|
|
73 |
decoded_sentence += " " + sampled_token
|
74 |
if sampled_token == "[end]":
|
75 |
break
|
76 |
+
return decoded_sentence[8:-6]
|
77 |
|
78 |
# ===== Enf of Keras ====
|
79 |
|