16 lines
521 B
Python
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)
|