aboutsummaryrefslogtreecommitdiffstats
path: root/learn_and_predict.py
diff options
context:
space:
mode:
Diffstat (limited to 'learn_and_predict.py')
-rw-r--r--learn_and_predict.py5
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)