From 5bcaa5edb67cab953737670ead39d0a3ebbf0bbd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= <patgal2@st.amu.edu.pl>
Date: Thu, 11 May 2023 00:57:13 +0200
Subject: [PATCH] Update 'plot.py'

---
 plot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plot.py b/plot.py
index 8ff24d8..38358e7 100644
--- a/plot.py
+++ b/plot.py
@@ -6,8 +6,8 @@ metrics = pandas.read_csv('./metrics.csv',
                               encoding='ISO-8859-1',
                               sep=',')
 
-builds = metrics[0]
-accuracy = metrics[1]
+builds = metrics[metrics.columns[0]]
+accuracy = metrics[metrics.columns[1]]
 
 
 plt.plot(builds,accuracy)