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

cvičení

Průměrné hodnoty segmentů NMF

Na závěr si vizuálně prozkoumáme průměrné hodnoty řešení se 3 segmenty sestaveného pomocí NMF a porovnáme ho s řešením K-means. Nejprve extrahujeme matici příznaků W, ze které určíme přiřazení zákazníků do segmentů – pro každého zákazníka vybereme segment odpovídající sloupci s nejvyšší hodnotou v této matici.

Knihovny pandas a seaborn jsou načteny jako pd, resp. sns. Surový dataset wholesale byl naimportován, natrénovaná instance NMF se 3 segmenty je k dispozici jako nmf a dataset components je načten jako DataFrame knihovny pandas.

Pokyny

100 XP
  • Vytvoř matici W tak, že jako data předáš transformované hodnoty a jako názvy sloupců použiješ index datasetu components.
  • Přiřaď hodnotu segmentu výběrem názvu sloupce s nejvyšší odpovídající hodnotou.
  • Vypočítej průměrné hodnoty sloupců pro každý segment.
  • Vykresli průměrné hodnoty jako heatmapu.