20 lines
528 B
Python
20 lines
528 B
Python
import matplotlib.pyplot as plt
|
|
|
|
def main():
|
|
accuracy = []
|
|
build_numbers = []
|
|
|
|
with open("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("RMSE")
|
|
plt.title("RMSE of the model over time")
|
|
plt.xticks(range(min(build_numbers), max(build_numbers) + 1))
|
|
plt.savefig("plot.png")
|
|
|
|
if __name__ == "__main__":
|
|
main() |