K-means – Salvar base original

[Alisson santos]

Bom dia, prof!
No caso do algortimo K-means quando finalizar todo processo como faço para concatenar a base original só com os resultados dos clusters e salvar em CSV?

[Denny Ceccon]

Olá Alisson,

Você pode usar o método transform ou fit_transform para gerar os nomes dos clusters, e depois adicionar uma coluna ao dataframe original com os rótulos dos grupos formados. Mais ou menos assim:

grupos = kmeans.fit_transform(X)
df['grupos'] = grupos
df.to_csv('kmeans.csv')

 

[Autor]

Posts