Plotting into CHART_DIR

Author: wrichert

ch05/log_reg_example.py

@@ -5,6 +5,9 @@
 #
 # It is made available under the MIT License
 
+import os
+from data import CHART_DIR
+
 import numpy as np
 from scipy.stats import norm
 
@@ -35,7 +38,7 @@ def lr_model(clf, X):
 pyplot.xlabel("feature value")
 pyplot.ylabel("class")
 pyplot.grid(True, linestyle='-', color='0.75')
-pyplot.savefig("log_reg_example_data.png", bbox_inches="tight")
+pyplot.savefig(os.path.join(CHART_DIR, "log_reg_example_data.png"), bbox_inches="tight")
 
 
 def lin_model(clf, X):
@@ -66,7 +69,7 @@ def lin_model(clf, X):
 pyplot.ylabel("class")
 pyplot.title("linear fit on additional data")
 pyplot.grid(True, linestyle='-', color='0.75')
-pyplot.savefig("log_reg_log_linear_fit.png", bbox_inches="tight")
+pyplot.savefig(os.path.join(CHART_DIR, "log_reg_log_linear_fit.png"), bbox_inches="tight")
 
 pyplot.figure(figsize=(10, 4))
 pyplot.xlim((-5, 20))
@@ -76,7 +79,7 @@ def lin_model(clf, X):
 pyplot.xlabel("feature value")
 pyplot.ylabel("class")
 pyplot.grid(True, linestyle='-', color='0.75')
-pyplot.savefig("log_reg_example_fitted.png", bbox_inches="tight")
+pyplot.savefig(os.path.join(CHART_DIR, "log_reg_example_fitted.png"), bbox_inches="tight")
 
 X = np.arange(0, 1, 0.001)
 pyplot.figure(figsize=(10, 4))
@@ -94,4 +97,4 @@ def lin_model(clf, X):
 pyplot.xlabel("P")
 pyplot.ylabel("log(odds) = log(P / (1-P))")
 pyplot.grid(True, linestyle='-', color='0.75')
-pyplot.savefig("log_reg_log_odds.png", bbox_inches="tight")
+pyplot.savefig(os.path.join(CHART_DIR, "log_reg_log_odds.png"), bbox_inches="tight")