ium_434704/create_chart.py
Wojciech Jarmosz 1490549284
Some checks failed
s434704-training/pipeline/head This commit looks good
s434704-evaluation/pipeline/head There was a failure building this commit
Fix chart generation
2021-05-15 01:42:32 +02:00

22 lines
443 B
Python

import sys
import os.path
import matplotlib
from matplotlib import pyplot as plt
x = []
y = []
if os.path.exists('bulk_metrics.txt'):
with open('bulk_metrics.txt') as file:
for line in file.readlines():
print(line.split(" "))
single_x, single_y = line.split(" ")
x.append(single_x)
y.append(single_y)
plt.plot(x, y)
plt.xlabel('Jobs')
plt.ylabel('RMSE')
plt.savefig('chart.png')