0
respostas

Problemas com TSNE

Olá, eu estou tentando usar o TSNE para agrupar o máximo de informação dos grupos em 2 dimensões mais estou tendo problemas, nesse caso o K-means só esta formando 3 grupos. Eu tentei fazer isso após a aula 2

from sklearn.manifold import TSNE

tsne = TSNE(n_components=2)
df_tsne = tsne.fit_transform(df)


grupador = KMeans(n_clusters= 4)
agrupador.fit(df_tsne)
labels = agrupador.labels_


print(labels)
fig = go.Figure()

fig.add_trace(go.Scatter(x= df_tsne[:,0], 
                           y= df_tsne[:, 1], mode= 'markers', 
                           marker= dict(color= agrupador.labels_.astype(np.float)),
                           text= labels))

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