objectif

Ajuster une régression linéaire avec summary.

code minimal

import statsmodels.api as sm
import numpy as np

X = np.column_stack([np.arange(10), np.ones(10)])
y = 2*X[:,0] + 1 + np.random.default_rng(0).normal(size=10)
model = sm.OLS(y, X).fit()
print(len(model.params))

utilisation

print(model.rsquared > 0.5)

variante(s) utile(s)

import statsmodels.api as sm
Xc = sm.add_constant(X[:,0])
print(sm.OLS(y, Xc).fit().params.shape[0])

notes

• add_constant ajoute l’ordonnée à l’origine.