diff options
Diffstat (limited to 'learn_and_predict.py')
-rw-r--r-- | learn_and_predict.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/learn_and_predict.py b/learn_and_predict.py index 961b190..4c094d7 100644 --- a/learn_and_predict.py +++ b/learn_and_predict.py | |||
@@ -101,7 +101,7 @@ def main(args): | |||
101 | model.summary() | 101 | model.summary() |
102 | 102 | ||
103 | adjuster = keras.callbacks.ReduceLROnPlateau( | 103 | adjuster = keras.callbacks.ReduceLROnPlateau( |
104 | monitor="val_acc", patience=5, verbose=1, factor=0.5, min_lr=0.0001 | 104 | monitor="val_accuracy", patience=5, verbose=1, factor=0.5, min_lr=0.0001 |
105 | ) | 105 | ) |
106 | 106 | ||
107 | history = model.fit( | 107 | history = model.fit( |
@@ -139,7 +139,8 @@ def main(args): | |||
139 | ] | 139 | ] |
140 | 140 | ||
141 | if not batch: | 141 | if not batch: |
142 | print(f"P@1: {precision_at_one/1000}, {precision_at_one} defs") | 142 | print(f"Supervised Retrieval {source_lang} - {target_lang}") |
143 | print(f"P@1: {precision_at_one/1000}") | ||
143 | else: | 144 | else: |
144 | with open("supervised.csv", "a") as f: | 145 | with open("supervised.csv", "a") as f: |
145 | writer = csv.writer(f) | 146 | writer = csv.writer(f) |