drawGraph.py

import matplotlib.pyplot as plt
import numpy as np
testData = []
trainData = []
costData = []
path = input("Insert path: ")
with open(path + "/testData.txt") as f:
    for line in f:
        
        editedLine = line.split(', ')
        editedLine.pop()
        for idx, x in enumerate(editedLine):
            testData.append(float(x))

with open(path + "/trainData.txt") as f:
    for line in f:
        editedLine = line.split(', ')
        editedLine.pop()
        for idx, x in enumerate(editedLine):
            trainData.append(float(x))

with open(path + "/costData.txt") as f:
    for line in f:
        editedLine = line.split(', ')
        editedLine.pop()
        for idx, x in enumerate(editedLine):
            costData.append(float(x)*100)

xaxis = []

for elem_idx in range(len(trainData)):
    xaxis.append(round(elem_idx, 4))

listof_YTicks = np.arange(0, 100, 5)
plt.yticks(listof_YTicks)

plt.plot(xaxis, trainData, label="Train accuracy")
plt.plot(xaxis, testData, label="Test accuracy")
plt.plot(xaxis, costData, label="Cost (x100)")
plt.xlabel("Epochs")
plt.ylabel("Accuracy % / Cost x 100")
plt.legend()
plt.show()