ium_z444510/5/predict.py
2023-06-27 19:56:40 +02:00

16 lines
521 B
Python

import tensorflow as tf
import pandas as pd
import numpy as np
if __name__ == '__main__':
test_data = pd.read_csv('../barcelona_weekends_datasets/test_set.csv')
x_test, y_test = test_data.iloc[:, :-1], test_data['guest_satisfaction_overall']
model = tf.keras.models.load_model('model_file.h5')
predictions = model.predict(x_test)
predictions = np.reshape(predictions, (-1,))
df_predictions = pd.DataFrame({'Predictions': predictions})
df_predictions.to_csv('predictions.csv', index=False)