Demosthene-OR
commited on
Commit
•
2bc4127
1
Parent(s):
e753237
Update main_dl.py
Browse files- main_dl.py +2 -3
main_dl.py
CHANGED
@@ -231,10 +231,9 @@ def load_all_data():
|
|
231 |
transformer_en_fr.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
|
232 |
transformer_fr_en.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
|
233 |
|
234 |
-
return
|
235 |
|
236 |
-
|
237 |
-
translation_en_fr, translation_fr_en, rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
|
238 |
|
239 |
|
240 |
def display_translation(n1, Lang,model_type):
|
|
|
231 |
transformer_en_fr.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
|
232 |
transformer_fr_en.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
|
233 |
|
234 |
+
return rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en
|
235 |
|
236 |
+
rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
|
|
|
237 |
|
238 |
|
239 |
def display_translation(n1, Lang,model_type):
|