PCAで遊ぶ．
pca.py
from sklearn.datasets import load_digits
from sklearn.decomposition import PCA
import matplotlib.pyplot as plt
## データの読み込み
digits = load_digits()
X = digits.data
y = digits.target
target_names = digits.target_names
## PCA
pca = PCA(n_components=2)
X_r = pca.fit(X).transform(X)
## colors
colors = [plt.cm.nipy_spectral(i/10., 1) for i in range(10)]
## plot
plt.figure()
for c, target_name in zip(colors, target_names):
plt.scatter(X_r[y == target_name, 0], X_r[y == target_name, 1], c=c, label = target_name)
plt.legend()
plt.title('PCA')
plt.show()
実行結果．
参考:
Scikit-learn PCAドキュメント
Scikit-learn PCAサンプル
東大富岡先生HP
PFIブログ