Untitled

 avatar
unknown
plain_text
a year ago
905 B
4
Indexable
from sklearn.cluster import KMeans
from sklearn.preprocessing import StandardScaler
from sklearn.decomposition import PCA
import plotly.express as px

# Предполагаем, что df_scaled и y_kmeans уже определены из предыдущего шага
# PCA для уменьшения размерности до 2D
pca_2d = PCA(n_components=2)
features_pca_2d = pca_2d.fit_transform(df_scaled)

# Создаем DataFrame для визуализации
df_pca_2d = pd.DataFrame(features_pca_2d, columns=['PC1', 'PC2'])
df_pca_2d['Cluster'] = y_kmeans

# Визуализация в 2D с использованием Plotly
fig = px.scatter(df_pca_2d, x='PC1', y='PC2', color='Cluster', 
                 title='2D PCA Visualization of Client Clusters',
                 labels={'Cluster': 'Cluster'},
                 color_continuous_scale=px.colors.qualitative.Set1)
fig.show()
Editor is loading...
Leave a Comment