figures

figures/plots.py

@@ -1,7 +1,9 @@
-import matplotlib.pyplot as plt
-plt.rcdefaults()
 import numpy as np
+import seaborn as sns
 import matplotlib.pyplot as plt
+
+sns.set(style="white", context="talk")
+
 plt.rcdefaults()
 
 DPI=600
@@ -39,17 +41,68 @@ plt.savefig(path+'.pdf')
 plt.savefig(path+'.png', dpi=DPI)
 print path
 
+# path="figures/mlfmm_minsky"
+# fig, ax = plt.subplots()
+# systems = ('1T', "160T", "1 GPU" ,"4 GPU")
+# mlfmm = (1.25e6, 4.71e4, 5.22e3, 1.29e3)
+# num = (44,44,44,44)
+# x_pos = np.arange(len(systems))
+# ax.bar(x_pos, [i/j for i,j in zip(mlfmm,num)], color='gray', log=True)
+# ax.set_xticks(x_pos)
+# ax.set_xticklabels(systems)
+# ax.set_ylabel("Per-MLFMM Execution Time (ms)")
+# plt.ylim([1, 1e5])
+# # ax.set_title('How fast do you want to go today?')
+# plt.savefig(path+'.pdf')
+# plt.savefig(path+'.png', dpi=DPI)
+# print path
+
 path="figures/mlfmm_minsky"
+# fig, ax = plt.subplots()
+# systems = ('1T', "160T", "1 GPU" ,"4 GPU")
+# mlfmm = [1.25e6, 4.71e4, 5.22e3, 1.29e3]
+# num = (44,44,44,44)
+# x_pos = [0,1,2,3]
+# sns.barplot(x_pos + x_pos, 
+#             [i/j for i, j in zip(mlfmm,num)] + [mlfmm[0] / i for i in mlfmm],
+#             ["Time", "Time", "Time", "Time", "Speedup", "Speedup", "Speedup", "Speedup"],
+#              palette="BuGn_d", ax=ax, log=True)
+
+width = 10
 fig, ax = plt.subplots()
 systems = ('1T', "160T", "1 GPU" ,"4 GPU")
-mlfmm = (1.25e6, 4.71e4, 5.22e3, 1.29e3)
+mlfmm = [1.25e6, 4.71e4, 5.22e3, 1.29e3]
 num = (44,44,44,44)
-x_pos = np.arange(len(systems))
-ax.bar(x_pos, [i/j for i,j in zip(mlfmm,num)], color='gray', log=True)
+x_pos = [0,1,2,3]
+sns.barplot(x_pos, 
+            [i/j for i, j in zip(mlfmm,num)],
+             color="DarkGray", ax=ax, log=True)
+ax.set_ylabel("Per-MLFMM Execution Time (ms)")
+
+ax2 = ax.twinx()
+# ax2.set_ylim((0,300))
+sns.barplot(x_pos, 
+            [mlfmm[0] / i for i in mlfmm],
+            color="LightGray", ax=ax2, log=True)
+ax2.set_ylabel("Speedup")
+
+
+new_width = 0.25
+for bar in ax.patches:
+    center = bar.get_x() + bar.get_width() / 2
+    bar.set_width(new_width)
+    bar.set_x(center - new_width)
+
+for bar in ax2.patches:
+    center = bar.get_x() + bar.get_width() / 2
+    bar.set_width(new_width)
+    bar.set_x(center)
+
+
 ax.set_xticks(x_pos)
 ax.set_xticklabels(systems)
-ax.set_ylabel("Per-MLFMM Execution Time (ms)")
-plt.ylim([1, 1e5])
+
+# plt.ylim([1, 1e5])
 # ax.set_title('How fast do you want to go today?')
 plt.savefig(path+'.pdf')
 plt.savefig(path+'.png', dpi=DPI)