Gradient decent for linear regression

gradient_descent.py

import numpy as np

def gradient_descent(x, y, alpha, rounds):
    z = np.ones(x.shapes[1])
    for _ in xrange(rounds):
        z = z - alpha * np.dot(np.dot(X, z) - y, X)
    return z