LinearRegression

Обычная линейная регрессия методом наименьших квадратов.

СигнатураLinearRegression(fit_intercept=True)

Строит линейную зависимость целевой переменной от признаков. После обучения коэффициенты доступны через coef_, а свободный член — через intercept_.

from sklearn.linear_model import LinearRegression

model = LinearRegression()
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
print(model.coef_, model.intercept_)
← Все записи: scikit-learn
Поддержать проект