from typing import List
import numpy as np
def income(st:List[int],inc:List[int],years:int)->int:
x =np.array(st).reshape(-1,1)
y =np.array(inc)
X =np.hstack((np.ones_like(x),x))
beta=np.linalg.inv(X.T @ X) @ X.T @ y
pred_income=beta[0]+beta[1]*years
return int(round(pred_income))
Пример
[22,9,27,11,21,10,20,7,26,5]
[46900,30500,51900,33200,45800,31600,44700,26600,51000,22400]
23
Вывод программы:47816
Правильный ответ:47815
Пример2
[22,9,27,11,21,10,20,7,26,5]
[46900,30500,51900,33200,45800,31600,44700,26600,51000,22400]
43
Вывод программы:73804
Правильный ответ:73803
сделай чтобы был верный ответ