1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning for Marketing in Python
Connected

cvičení

Prozkoumání koeficientů modelu

Teď se na výkon modelu podíváš z jiného úhlu – a to pouze na trénovacích datech. V poslední lekci jsi se dozvěděl/a, že ne všechny koeficienty modelu jsou statisticky významné, a proto bychom měli prozkoumat tabulku souhrnu modelu. Tuto funkcionalitu naštěstí nabízí knihovna statsmodels. Jakmile tabulku souhrnu zobrazíš, podívej se, které proměnné mají p-hodnotu nižší než 0,05 (tj. nižší než 5 %) – to znamená, že jejich koeficient je statisticky významný.

Trénovací příznaky jsou načteny jako train_X a cílová proměnná jako train_Y, která byla převedena na pole numpy.

Pokyny

100 XP
  • Importuj modul statsmodels.api.
  • Inicializuj instanci modelu na trénovacích datech pomocí funkce OLS().
  • Natrénuj model.
  • Zobraz souhrn modelu pomocí metody .summary().