Apprentissage supervisé avec scikit-learn
George Boorman
Core Curriculum Manager, DataCamp
$y = ax + b$
La régression linéaire simple utilise une caractéristique
$y$ = cible
$x$ = caractéristique unique
$a$, $b$ = paramètres/coefficients du modèle : pente, ordonnée à l’origine
Comment choisir $a$ et $b$ ?
Définissez une fonction d’erreur pour une ligne donnée
Choisissez la ligne qui minimise la fonction d’erreur
Fonction d’erreur = fonction de perte = fonction de coût
$RSS = $ $\displaystyle\sum_{i=1}^{n}(y_i-\hat{y_i})^2$
Moindres carrés ordinaires (MCO) : minimiser RSS
$$ y = a_{1}x_{1} + a_{2}x_{2} + b$$
$$ y = a_{1}x_{1} + a_{2}x_{2} + a_{3}x_{3} +... + a_{n}x_{n}+ b$$
from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
reg_all = LinearRegression()
reg_all.fit(X_train, y_train)
y_pred = reg_all.predict(X_test)
$R^2$ : quantifie la variance des valeurs cibles expliquée par les caractéristiques
$R^2$ élevé :
reg_all.score(X_test, y_test)
0.356302876407827
$MSE = $ $\displaystyle\frac{1}{n}\sum_{i=1}^{n}(y_i-\hat{y_i})^2$
$RMSE = $ $\sqrt{MSE}$
from sklearn.metrics import root_mean_squared_error
root_mean_squared_error(y_test, y_pred)
24.028109426907236
Apprentissage supervisé avec scikit-learn