From a2b9b274b2229756a767424e1c26d967dd8408b7 Mon Sep 17 00:00:00 2001
From: s434695 <tomgrz3@st.amu.edu.pl>
Date: Sun, 16 May 2021 23:02:31 +0200
Subject: [PATCH] evaluation

---
 evaluate.py   |  1 -
 evaluation.py | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)
 delete mode 100644 evaluate.py
 create mode 100644 evaluation.py

diff --git a/evaluate.py b/evaluate.py
deleted file mode 100644
index e0369f7..0000000
--- a/evaluate.py
+++ /dev/null
@@ -1 +0,0 @@
-print('test')
\ No newline at end of file
diff --git a/evaluation.py b/evaluation.py
new file mode 100644
index 0000000..1f10f2a
--- /dev/null
+++ b/evaluation.py
@@ -0,0 +1,38 @@
+import pandas as pd
+import numpy as np
+from tensorflow import keras
+import matplotlib.pyplot as plt
+from sklearn.metrics import mean_squared_error
+
+
+vgsales_model = 'vgsales_model.h5'
+model = keras.models.load_model(vgsales_model)
+
+vgsales_test = pd.read_csv('test.csv')
+
+vgsales_test['Nintendo'] = vgsales_test['Publisher'].apply(lambda x: 1 if x=='Nintendo' else 0)
+
+X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
+y_test = vgsales_test[['Nintendo']]
+
+predictions = model.predict(X_test)
+
+error = mean_squared_error(y_test, predictions)
+print('Error: ', error)
+
+with open('results.txt', 'a') as f:
+    f.write(str(error) + "\n")
+
+
+with open('results.txt', 'r') as f:
+    lines = f.readlines()
+
+
+fig = plt.figure(figsize=(5,5))
+chart = fig.add_subplot()
+chart.set_ylabel("MSE")
+chart.set_xlabel("Build")
+x = np.arange(0, len(lines), 1)
+y = [float(x) for x in lines]
+plt.plot(x, y, "ro")
+plt.savefig("plot.png")
\ No newline at end of file