Линейная регрессия · прогноз на 3 месяца
данные хранятся локально
| Месяц | Продажи (₽) |
|---|
# линейная регрессия: y = a·x + b
def predict(months, sales, x):
n = len(months)
x_mean = sum(months)/n
y_mean = sum(sales)/n
a = sum((m-x_mean)*(s-y_mean) for m,s in zip(months,sales)) / sum((m-x_mean)**2 for m in months)
b = y_mean - a*x_mean
return [a*xi + b for xi in x]