Fixes some error bars in benchmark plots.

noripyt · Nov 23, 2014 · 3eb8dcd · 3eb8dcd
1 parent ee33a6c
commit 3eb8dcd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/benchmark.py b/benchmark.py
@@ -125,14 +125,15 @@ def plot(self):
         for db in self.means.columns.levels[0]:
             plt.figure()
             self.means[db].plot(
-                kind='barh', xerr=self.stds[db], title=db, xlim=xlim,
-                figsize=(15, 10), subplots=True, layout=(4, 2),
+                kind='barh', xerr=self.stds[db], title=db,
+                xlim=xlim, figsize=(15, 10), subplots=True, layout=(4, 2),
                 sharey=True, legend=False)
             plt.savefig(os.path.join(RESULTS_PATH, '%s_detail.svg' % db))
 
             plt.figure()
             self.means[db].mean(axis=1).plot(
-                kind='barh', xerr=self.stds[db], title=db, xlim=xlim)
+                kind='barh', xerr=self.stds[db].mean(axis=1), title=db,
+                xlim=xlim)
             plt.savefig(os.path.join(RESULTS_PATH, '%s.svg' % db))