1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Supervised Learning with scikit-learn
Connected

Cvičení

Trénování a predikce pro regresi

Teď, když víš, jak lineární regrese funguje, je tvým úkolem vytvořit model mnohonásobné lineární regrese s využitím všech příznaků z datasetu sales_df, který je pro tebe předem načtený. Jako připomínka, tady jsou první dva řádky:

     tv        radio      social_media    sales
1    13000.0   9237.76    2409.57         46677.90
2    41000.0   15886.45   2913.41         150177.83

Tento model pak použiješ k předpovězení prodejů na základě hodnot testovacích příznaků.

LinearRegression a train_test_split jsou pro tebe předem načteny z příslušných modulů.

Pokyny

100 XP
  • Vytvoř X jako pole obsahující hodnoty všech příznaků z sales_df a y obsahující všechny hodnoty ze sloupce "sales".
  • Vytvoř instanci modelu lineární regrese.
  • Natrénuj model na trénovacích datech.
  • Vytvoř y_pred tak, že pomocí testovacích příznaků předpovíš hodnoty sales.