Advertisement
Guest User

Untitled

a guest
Sep 25th, 2017
61
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.53 KB | None | 0 0
  1. def step_gradient(c_current, m_current, points, learning_rate):
  2. # Gradient descent
  3. c_gradient = 0
  4. m_gradient = 0
  5. N = float(len(points))
  6. # Iterate
  7. for i in range(0, len(points)):
  8. x = points[i, 0]
  9. y = points[i, 1]
  10. c_gradient += -(2 / N) * (y - ((m_current * x) + c_current))
  11. m_gradient += -(2 / N) * x * (y - ((m_current * x) + c_current))
  12. # Update m and c
  13. new_c = c_current - (learning_rate * c_gradient)
  14. new_m = m_current - (learning_rate * m_gradient)
  15. return [new_c, new_m]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement