ium_464913/plot.py
2024-05-04 16:39:48 +02:00

25 lines
581 B
Python

import matplotlib.pyplot as plt
def main():
accuracy = []
build_numbers = []
with open("evaluation/metrics.txt") as f:
for line in f:
accuracy.append(float(line.split(",")[0]))
build_numbers.append(int(line.split(",")[1]))
plt.plot(build_numbers, accuracy)
plt.xlabel("Build Number")
plt.ylabel("Accuracy")
plt.title("Accuracy of the model over time")
plt.xticks(range(min(build_numbers), max(build_numbers) + 1))
plt.show()
plt.savefig("evaluation/accuracy.png")
if __name__ == "__main__":
main()