Create prediction.py

KOSASIH · May 11, 2024 · 5ba03f6 · 5ba03f6
1 parent fa972a4
commit 5ba03f6
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