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:39:44 +02:00
parent f5f69a488b
commit 4e825aa649
2 changed files with 19 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
IUM_05-predict.py
View File

@ -4,15 +4,16 @@ import tensorflow as tf
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']
model = tf.keras.models.load_model('beer_review_sentiment_model.h5')
tokenizer = tf.keras.preprocessing.text.Tokenizer(num_words=10000)
tokenizer.fit_on_texts(X_test)
X_test_seq = tokenizer.texts_to_sequences(X_test)
X_test_pad = tf.keras.preprocessing.sequence.pad_sequences(X_test_seq, maxlen=100)
predictions = model.predict(X_test_pad)
np.savetxt('beer_review_sentiment_predictions.csv', predictions, delimiter=',', fmt='%.10f')
results = pd.DataFrame({'Predictions': predictions.flatten(), 'Actual': y_test})
results.to_csv('beer_review_sentiment_predictions.csv', index=False)

30
IUM_06-metrics.py
View File

@ -1,19 +1,19 @@
from sklearn.metrics import confusion_matrix
import pandas as pd
import sys
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, mean_squared_error
from math import sqrt
# Load the predictions data
data = pd.read_csv('beer_review_sentiment_predictions.csv')
y_pred = data['Predictions']
y_test = data['Actual']
def main():
y_test = pd.read_csv("beer_reviews_test.csv")
y_pred = pd.read_csv("beer_review_sentiment_predictions.csv", header=None)
build_number = sys.argv[1]
# Calculate metrics
accuracy = accuracy_score(y_test, y_pred.round())
precision, recall, f1, _ = precision_recall_fscore_support(y_test, y_pred.round(), average='micro')
rmse = sqrt(mean_squared_error(y_test, y_pred))
cm = confusion_matrix(y_test, y_pred)
accuracy = cm[1, 1] / (cm[1, 0] + cm[1, 1])
with open(r"beer_metrics.txt", "a") as f:
f.write(f"{accuracy},{build_number}\n")
if __name__ == "__main__":
main()
print(f'Accuracy: {accuracy}')
print(f'Micro-avg Precision: {precision}')
print(f'Micro-avg Recall: {recall}')
print(f'F1 Score: {f1}')
print(f'RMSE: {rmse}')