diff --git a/machine_learning/prediction.py b/machine_learning/prediction.py index 594e6cc..540465a 100644 --- a/machine_learning/prediction.py +++ b/machine_learning/prediction.py @@ -1,5 +1,6 @@ import tensorflow as tf + def predict(model, data): # Make predictions using the trained model predictions = model.predict(data)