Create prediction.py

KOSASIH · May 11, 2024 · 78361d3 · 78361d3
1 parent 2bba486
commit 78361d3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/machine_learning/prediction.py b/machine_learning/prediction.py
@@ -0,0 +1,7 @@
+import tensorflow as tf
+
+def predict(model, data):
+    # Make predictions using the trained model
+    predictions = model.predict(data)
+
+    return predictions