Как написать функцию для вычисления оценки квадрата ошибки в линейной регрессии? - PullRequest
Новая
       15

Как написать функцию для вычисления оценки квадрата ошибки в линейной регрессии?

0 голосов
/ 09 апреля 2020

Я пытаюсь написать функцию, которая принимает в качестве входных данных матрицу элементов X, наблюдаемый выходной вектор y и массив параметров theta, а в качестве выходных данных дает общую стоимость (т.е. среднеквадратическую ошибку). Я получаю синтаксическую ошибку в последней строке (которая должна возвращать выходные данные), но я не могу выяснить причину. 

def computeCost(X, y, theta):

    m = y.size  # number of training examples

    J = 0

    for i in range(m):
        error = ((theta[0])+(theta[1]*X[i])-(y[i])**2
        J = J + error


    return J*(1/(2*m))
...