1. Learn
    2. /
  2. Cursuri
    3. /
  3. Învățare supravegheată cu scikit-learn
Connected

exercițiu

Construirea unui model de regresie liniară

Acum că ai creat vectorii de caracteristici și de țintă, vei antrena un model de regresie liniară pe toate valorile disponibile.

Deoarece scopul este să evaluezi relația dintre caracteristici și valorile țintă, nu este nevoie să împarți datele în seturi de antrenament și de testare.

X și y au fost preîncărcate astfel:

y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)

Instrucțiuni

100 XP
  • Importă LinearRegression.
  • Instanțiază un model de regresie liniară.
  • Generează predicții pentru valorile vânzărilor folosind X și stochează rezultatul în predictions.