Линейная регрессия | Реализация формулы в python

class LinearRegression: def __init__(self): self.k=None self.b=None def train(self,x_list,y_list): n=len(x_list) sum_xy=sum(x*y for x,y in zip(x_list,y_list)) sum_x=sum(x for x in x_list) sum_y=sum(y for y in y_list) sum_x2=sum(x*x for x in x_list) self.k = (n*sum_xy-sum_x*sum_y)/(n*sum_x2-sum_x**2) self.b = (sum_y-self.k*sum_x)/n def predict(self,x_predictions): y_predictions = [self.k*x_dot+self.b for x_dot in x_predictions] return y_predictions x_list = [-2, -1, 0, 1, 2] y_list = [16, 13, 10, 7, 4] x_predictions = [3, -3] Ожидаемый вывод: 1.0 19.0 model=LinearRegression() model.train(x_list,y_list) y_predictions=model.predict(x_predictions) for y_pred in y_predictions: print(y_pred)