Solucionado (ver solução)
Solucionado
(ver solução)
4
respostas

Duvida sobre o fit do KMeans

Ao realizar esse comando:

modelo = KMeans(n_clusters = 3)
modelo.fit(generos_alterados)

a função fit do KMeans, manipula e acrescenta informações sobre a clusterização nesse variavel/objeto (generos_alterados)?

Minha duvida decorre no fato desse objeto ser utilizado posteriormente para redimensionamento e plotagem, logo ele precisaria ter informações sobre os clusters.

4 respostas

Bruno, tudo bom?

De forma macro, os dados do objeto serão classificados conforme o modelo utilizado (no caso, com 3 clusteres). Realmente, você não terá esse dado no objeto generos_alterados, mas para plotar um gráfico vc não precisará dele afinal isso será feito com os dados.

Obrigado pela resposta. Mas se puder me ajudar mais um pouco:

Mas ao plotar o gráfico precisamos colorir os pontos, baseados nos grupos. Como que o gráfico sabe a qual cluster pertence o ponto?

solução!

Exatamente pelo que você falou Bruno. Imagine que ao passar pelo modelo são formados 3 grupos de dados conforme aqueles 3 clusteres definidos. Você não tem o valor 3 como conversamos, mas terá 3 agrupamentos de dados para plotar o gráfico.

Obrigado, Ronald.

Quer mergulhar em tecnologia e aprendizagem?

Receba a newsletter que o nosso CEO escreve pessoalmente, com insights do mercado de trabalho, ciência e desenvolvimento de software