@@ -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")
@@ -41,11 +41,11 @@
     import twitter
 except ImportError:
     print("""\
-You need to install python-twitter.
-On Linux:
-    sudo pip install python-twitter
-On Windows:
-    Follow instructions at http://code.google.com/p/python-twitter/ """)
+You need to install python-twitter:
+    pip install python-twitter
+If pip is not found you might have to install it using easy_install.
+If it does not work on your system, you might want to follow instructions 
+at http://code.google.com/p/python-twitter/ """)
 
     sys.exit(1)