Update 22_linear_regression_direct_inverse.py

Author: thomasnield

03_linear_algebra/22_linear_regression_direct_inverse.py

@@ -21,3 +21,15 @@
 
 # Predict against the y-values 
 y_predict = X_1.dot(b)
+
+
+# plot
+import matplotlib.pyplot as plt
+import numpy as np
+# show in chart
+X = np.array([p.x for p in df.itertuples()])
+Y = np.array([p.y for p in df.itertuples()])
+
+plt.plot(X, Y, 'o') # scatterplot
+plt.plot(X, m*X+b) # line
+plt.show()