Pythonで学ぶマーケティングのための機械学習
Karolis Urbonas
Head of Analytics & Science, Amazon
K-means によるセグメンテーション(k クラスタ):
from sklearn.cluster import KMeanskmeans=KMeans(n_clusters=k)kmeans.fit(wholesale_scaled_df)wholesale_kmeans4 = wholesale.assign(segment = kmeans.labels_)
NMF によるセグメンテーション(k クラスタ):
from sklearn.decomposition import NMF
nmf = NMF(k)
nmf.fit(wholesale)
components = pd.DataFrame(nmf.components_, columns=wholesale.columns)
セグメント割当の抽出:
segment_weights = pd.DataFrame(nmf.transform(wholesale, columns=components.index)
segment_weights.index = wholesale.index
wholesale_nmf = wholesale.assign(segment = segment_weights.idxmax(axis=1))
k)が必要k の決め方は 1) 数理 2) テスト&ラーンk を反復k でクラスタリング実行k の二乗誤差和(SSE)を算出SSE を k に対してプロットし、「肘」=誤差減少の逓減点を特定sse = {}
for k in range(1, 11):
kmeans=KMeans(n_clusters=k, random_state=333)
kmeans.fit(wholesale_scaled_df)
sse[k] = kmeans.inertia_
plt.title('Elbow criterion method chart')
sns.pointplot(x=list(sse.keys()), y=list(sse.values()))
plt.show()
k の前後で複数値でセグメントを作成Pythonで学ぶマーケティングのための機械学習