From f416349645528422f99e1b7f61f300d48293d577 Mon Sep 17 00:00:00 2001 From: szymonj98 Date: Sun, 1 May 2022 19:50:46 +0200 Subject: [PATCH] plot --- evaluation.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/evaluation.py b/evaluation.py index 10201e6..3e59ebe 100644 --- a/evaluation.py +++ b/evaluation.py @@ -4,6 +4,7 @@ import pandas as pd import numpy as np import csv from sklearn.model_selection import train_test_split +from matplotlib import pyplot as plt x_test = pd.read_csv('xtest.csv') y_test = pd.read_csv('ytest.csv') @@ -13,4 +14,13 @@ model = tf.keras.models.load_model('./model') res = model.evaluate(x_test, y_test,verbose=0) with open('evaluation_acuraccy.txt', 'a+') as f: - f.write(str(res[1])+'\n') \ No newline at end of file + f.write(str(res[1])+'\n') +with open('evaluation_acuraccy.txt') as f: + scores = [float(line) for line in f if line] + print(scores) + builds = list(range(1, len(scores) + 1)) + plot = plt.plot(builds, scores) + plt.xlabel('Build') + plt.xticks(range(1, len(scores) + 1)) + plt.ylabel('Accuraccy') + plt.show() \ No newline at end of file