s464979/ium_464979
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

IUM_06

Browse Source
This commit is contained in:
AWieczarek 2024-05-06 19:52:46 +02:00
parent 4ee651cb5d
commit 8976e3ad75
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
IUM_05-predict.py
View File

@ -2,28 +2,24 @@ import pandas as pd
import numpy as np
import tensorflow as tf
# Load the test data
test_data = pd.read_csv('./beer_reviews_test.csv')
X_test = test_data[['review_aroma', 'review_appearance', 'review_palate', 'review_taste']]
y_test = test_data['review_overall']
# Load the model
model = tf.keras.models.load_model('beer_review_sentiment_model.h5')
tokenizer = tf.keras.preprocessing.text.Tokenizer(num_words=10000)
X_test_list = X_test.values.tolist()
tokenizer.fit_on_texts(X_test_list)
X_test_seq = tokenizer.texts_to_sequences(X_test_list)
X_test_pad = tf.keras.preprocessing.sequence.pad_sequences(X_test_seq, maxlen=100)
predictions = model.predict(X_test_pad)
# Make predictions
predictions = model.predict(X_test)
# Check the shape of the predictions
print(f'Predictions shape: {predictions.shape}')
# If predictions have more than one dimension, select only one dimension
if len(predictions.shape) > 1:
predictions = predictions[:, 0]
# Save predictions and actual data to a CSV file
results = pd.DataFrame({'Predictions': predictions, 'Actual': y_test})
results.to_csv('beer_review_sentiment_predictions.csv', index=False)