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) => {