evaluation metrics plot

This commit is contained in:
Szymon Bartanowicz 2024-05-15 00:57:08 +02:00
parent 8d92919488
commit 4a7fe811f5

View File

@ -4,7 +4,7 @@ def main():
accuracy = [] accuracy = []
build_numbers = [] build_numbers = []
with open("maetrics.txt") as f: with open("metrics.txt") as f:
for line in f: for line in f:
accuracy.append(float(line.split(",")[0])) accuracy.append(float(line.split(",")[0]))
build_numbers.append(int(line.split(",")[1])) build_numbers.append(int(line.split(",")[1]))
@ -14,8 +14,6 @@ def main():
plt.ylabel("RMSE") plt.ylabel("RMSE")
plt.title("RMSE of the model over time") plt.title("RMSE of the model over time")
plt.xticks(range(min(build_numbers), max(build_numbers) + 1)) plt.xticks(range(min(build_numbers), max(build_numbers) + 1))
plt.show()
plt.savefig("plot.png") plt.savefig("plot.png")
if __name__ == "__main__": if __name__ == "__main__":