K-means – Salvar base original

Home › Fóruns › Fórum Machine Learning e Data Science com Python de A à Z › K-means – Salvar base original

Marcado: 

Visualizando 2 posts - 1 até 2 (de 2 do total)
  • Autor
    Posts
  • 26 de novembro de 2021 às 11:50 #32399
    Alisson santos
    Participante

      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?

      29 de novembro de 2021 às 12:28 #32442
      Denny Ceccon
      Moderador

        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
      Visualizando 2 posts - 1 até 2 (de 2 do total)
      • Você deve fazer login para responder a este tópico.