s464962/ium_464962
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix predict.py

Browse Source
This commit is contained in:
Krzysztof Raczyński 2024-05-13 21:03:34 +02:00
parent 2bbee09bd5
commit 4b3fdb7c23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
predict.py
View File

@ -4,7 +4,7 @@ from tensorflow.keras import Sequential
import tensorflow as tf import tensorflow as tf
from sklearn.preprocessing import MinMaxScaler 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) test_data.dropna(inplace=True)
y_test = test_data['sellingprice'].astype(np.float32) 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['condition'] = scaler_X.fit_transform(X_test[['condition']])
X_test = pd.get_dummies(X_test, columns=['transmission']) 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) y_pred_scaled = model.predict(X_test)