Fixing Errors

vivek3141 · Mar 25, 2019 · a0eb90d · a0eb90d
1 parent b5d82af
commit a0eb90d
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/ml/linear_regression/lreg.py b/ml/linear_regression/lreg.py
@@ -55,17 +55,13 @@ def fit(self, data, labels, lr=0.01, graph=False, steps=1000, init_theta=(1, 1))
         :param init_theta: Initial theta values
         :return: Cost of the model
         """
-        theta = init_theta
-        x = data
-        y = labels
-        n = len(x)
         self.theta = lin_reg.fit(list(data), list(
             labels), lr, steps, list(init_theta), len(data))
 
         if graph:
-            x_line = np.array([min(x), max(x)])
+            x_line = np.array([min(data), max(data)])
             y_line = self.theta[0] * x_line + self.theta[1]
-            plt.scatter(x, y, c="RED")
+            plt.scatter(data, labels, c="RED")
             plt.plot(x_line, y_line)
             plt.show()
         return