add plt.show()

probml · Mar 4, 2021 · bd21b99 · bd21b99
1 parent 160122b
commit bd21b99
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/scripts/__pycache__/pyprobml_utils.cpython-38.pyc b/scripts/__pycache__/pyprobml_utils.cpython-38.pyc
diff --git a/scripts/ab_test_demo.py b/scripts/ab_test_demo.py
@@ -75,6 +75,7 @@ def eprofit_total(N, n1, n2, s, mu, sigma):
 plt.axhline(eprofit_opt)
 plt.text(nopt, eprofit_opt, 'n*={:0.1f}'.format(nopt))
 savefig('ab_profit.png')
+plt.show()
 
 plt.figure();
 plt.plot(ns, error_rate)
@@ -83,4 +84,5 @@ def eprofit_total(N, n1, n2, s, mu, sigma):
 plt.axvline(nopt)
 plt.axhline(error_rate_opt)
 plt.text(nopt, error_rate_opt, 'n*={:0.1f}'.format(nopt))
-savefig('ab_error.png')
+savefig('ab_error.png')
+plt.show()
diff --git a/scripts/gauss_plot.py b/scripts/gauss_plot.py
@@ -2,9 +2,9 @@
 
 import numpy as np
 import matplotlib.pyplot as plt
-import os
-figdir = "../figures"
-def save_fig(fname): plt.savefig(os.path.join(figdir, fname))
+
+
+import pyprobml_utils as pml
 
 from scipy.stats import norm
 
@@ -14,11 +14,11 @@ def save_fig(fname): plt.savefig(os.path.join(figdir, fname))
 fig, ax = plt.subplots()
 ax.plot(X, rv.pdf(X))
 plt.title("Gaussian pdf")
-save_fig("gaussian1d.pdf")
+pml.save_fig("gaussian1d.pdf")
 plt.show()
 
 fig, ax = plt.subplots()
 ax.plot(X, rv.cdf(X))
 plt.title("Gaussian cdf")
-save_fig("gaussianCdf.pdf")
+pml.save_fig("gaussianCdf.pdf")
 plt.show()