changes to train file

utterworks · Mar 9, 2021 · 439739c · 439739c
1 parent 71bee36
commit 439739c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/container/bert/train b/container/bert/train
@@ -213,6 +213,10 @@ def train():
             optimizer_type=hyperparameters["optimizer_type"],
         )
 
+        results = learner.validate(return_preds=True)
+        logger.info("y_pred: {}".format(json.dumps(results["y_pred"].tolist())))
+        logger.info("y_true: {}".format(json.dumps(results["y_true"].tolist())))
+
         # save model and tokenizer artefacts
         learner.save_model()