Skip to content

Commit

Permalink
update plot rmse script
Browse files Browse the repository at this point in the history
  • Loading branch information
solomonik committed Oct 5, 2019
1 parent fa50285 commit 59c7dc9
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions test/plot_rmse_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,22 +28,25 @@ def plot_each():
plt.plot(time[i], rmse[i], 'ro')
plt.plot(time[i], rmse[i])
plt.title(method[i])
plt.xlabel('Time')
plt.ylabel('RMSE')
plt.xlabel('Time (seconds)')
plt.ylabel('RMSE (average error)')
plt.show()

def plot_together():
plt.figure(figsize=(7,4))
colors = ['r', 'b', 'g', 'brown']
linestyles = ['--', '-.', ':', '-']
markers = ['s', 'x', '^', 'd']
for i in range(len(method)):
for i in range(len(method))[::-1]:
plt.plot(time[i], rmse[i], color=colors[i], linestyle=linestyles[i], marker=markers[i], label=method[i])
plt.xlabel('Time')
plt.ylabel('RMSE')
plt.title('Tensor Completion using Cyclops on 4096 Cores of Stampede2')
plt.xlabel('Time (seconds)')
plt.ylabel('RMSE (average entrywise model error)')
plt.yscale("log")
plt.legend(loc='best')
plt.grid(linestyle='--')
plt.show()

parse_output('out.txt')
parse_output('pp.bench.ss.N64.func.out')
#plot_each()
plot_together()

0 comments on commit 59c7dc9

Please sign in to comment.