From 0445bc9d821def9f66676678b038773ab86f558c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Wed, 10 May 2023 22:59:58 +0200 Subject: [PATCH] Update 'predict.py' --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 290b71f..ccba156 100644 --- a/predict.py +++ b/predict.py @@ -7,7 +7,7 @@ from sklearn.model_selection import train_test_split import tensorflow -reloaded = tensorflow.keras.models.load_model('.') +reloaded = tensorflow.keras.models.load_model('saved_model.pb') x_to_test = pandas.read_csv('./X_test.csv') y_to_test = pandas.read_csv('./Y_test.csv')