from sklearn import decomposition
import matplotlib.pyplot as plt
import numpy as np
# import seaborn #可视化
from mpl_toolkits.mplot3d import Axes3D
%matplotlib notebook
data = np.loadtxt('batch10_random_train.txt')
data_x = data[:,6:]
data_y = data[:,:6]
data_y = [np.argmax(one_hot) for one_hot in data_y] #转成一类
pca = decomposition.PCA(n_components=3)
new_X = pca.fit_transform(data_x)
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.scatter(new_X[:, 0], new_X[:, 1], new_X[:, 2], c=data_y, cmap=plt.cm.spectral)#生成几种不同的颜色
plt.show()
pca降维
最新推荐文章于 2022-05-07 18:57:59 发布