• 17
    いいね
  • 0
    コメント
この記事は最終更新日から1年以上が経過しています。

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()

実行結果.

Untitled.png

参考:
Scikit-learn PCAドキュメント
Scikit-learn PCAサンプル
東大富岡先生HP
PFIブログ