From 621422979ae71e25ef583ac82bbf6419d68e28bb Mon Sep 17 00:00:00 2001
From: Kieran McAuliffe <69212477+trian-gles@users.noreply.github.com>
Date: Mon, 1 Jul 2024 15:54:49 +0200
Subject: [PATCH] training_done message emitted

---
 tf.js | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tf.js b/tf.js
index 855e2a9..7b19d32 100644
--- a/tf.js
+++ b/tf.js
@@ -31,6 +31,8 @@ maxApi.addHandler("train", (epochs) => {
 
   // Train the model using the data.
   model.fit(xs, ys, {epochs});
+
+  maxApi.outlet("training_done");
 });
 
 maxApi.addHandler("dataPoint", (...data) => {