Update Dockerfile

Dockerfile

@@ -4,7 +4,7 @@ RUN apt update && apt install -y python3-pip
 RUN apt install unzip
 RUN apt install bc
 
-RUN pip3 install kaggle pandas scikit-learn torch
+RUN pip3 install kaggle pandas scikit-learn torch matplotlib
 
 WORKDIR /app
 

plot.py

@@ -9,9 +9,9 @@ for line in f:
   if(parts[0] == 'Accuracy:'):
     accuracy.append(float(parts[1]))
 
-y_values = np.arange(1, len(accuracy) + 1)
+build_numbers = np.arange(1, len(accuracy) + 1)
 
-plt.plot(y_values, accuracy, marker='o', linestyle='-', color='b')
+plt.plot(accuracy, build_numbers, marker='o', linestyle='-', color='b')
 plt.xlabel('Build Number')
 plt.ylabel('Accuracy')
 plt.title('Accuracy Plot')