From 4b3fdb7c23cf8dc32deb3b1e3c2510eea8e7e44c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Krzysztof=20Raczy=C5=84ski?= Date: Mon, 13 May 2024 21:03:34 +0200 Subject: [PATCH] fix predict.py --- predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/predict.py b/predict.py index 3059f66..b674665 100644 --- a/predict.py +++ b/predict.py @@ -4,7 +4,7 @@ from tensorflow.keras import Sequential import tensorflow as tf from sklearn.preprocessing import MinMaxScaler -test_data = pd.read_csv('./data/car_prices_test.csv') +test_data = pd.read_csv('./car_prices_test.csv') test_data.dropna(inplace=True) y_test = test_data['sellingprice'].astype(np.float32) @@ -17,7 +17,7 @@ scaler_X = MinMaxScaler() X_test['condition'] = scaler_X.fit_transform(X_test[['condition']]) X_test = pd.get_dummies(X_test, columns=['transmission']) -model = tf.keras.models.load_model('car_prices_predict_model.h5') +model = tf.keras.models.load_model('./car_prices_predict_model.h5') y_pred_scaled = model.predict(X_test)