Линейная регрессия: предсказание выручки
Medium
scikit-learn
ML
Wildberries
Условие задачи
Дано: DataFrame `data` с колонками ad_spend, visits, revenue -- 20 строк.
Обучите линейную регрессию revenue ~ ad_spend + visits и получите коэффициенты при признаках, свободный член и R² на обучающих данных.
Сохраните dict {"coefficients": list(round 2), "intercept": round(2), "r2": round(4)} в `result`.
Темы
linear_regression r2 prediction
Подсказки
- LinearRegression().fit(X, y) обучает модель
- model.coef_ -- коэффициенты, model.intercept_ -- свободный член
- model.score(X, y) -- R2
Открыть задачу в тренажёре →
← Все Python-задачи