Exemplo n.º 1
0
 def run_linear_regression(self, x_axis, y_axis):
     self.x_axis = self.headers[x_axis]
     self.y_axis = self.headers[y_axis]
     self.slope = MathsUtil.estimate_slope(self.data[self.x_axis], self.data[self.y_axis])
     self.y_intercept = MathsUtil.estimate_y_intercept(self.data[self.x_axis], self.data[self.y_axis], self.slope)
     self.predicted_y = MathsUtil.num_plus_arr(self.y_intercept, MathsUtil.num_by_arr(self.slope, self.data[self.x_axis]))
 def test_estimate_y_intercept(self):
     self.assertEqual(MathsUtil.estimate_y_intercept([1, 2, 3], [1, 2, 3], 1), 0)