PCA demo(2) 手动设定n_components

from sklearn.linear_model import  SGDRegressor
from sklearn import datasets
from sklearn.model_selection import train_test_split#将数据分为测试集和训练集
from sklearn.preprocessing import StandardScaler
from sklearn.decomposition import PCA
from sklearn.neighbors import KNeighborsClassifier
import matplotlib.pyplot as plt
import numpy as np


digits=datasets.load_digits()
x=digits.data
y=digits.target

X_train,X_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=666)

knn=KNeighborsClassifier()
knn.fit(X_train,y_train)
print(knn.score(X_test, y_test))


pca=PCA(n_components=X_train.shape[1])
pca.fit(X_train)
print(pca.explained_variance_ratio_)


plt.plot([i for i in range(X_train.shape[1])],[np.sum(pca.explained_variance_ratio_[:i+1]) for i in range(X_train.shape[1])])
plt.show()