PCA to narzędzie do zmniejszania wymiarów, a nie klasyfikator. W Scikit-Learn wszystkie klasyfikatory i estymatory mają predict
metodę, której nie ma PCA . Musisz dopasować klasyfikator do danych transformowanych PCA. Scikit-Learn ma wiele klasyfikatorów. Oto przykład użycia drzewa decyzyjnego na danych transformowanych PCA. Wybrałem klasyfikator drzewa decyzyjnego, ponieważ działa on dobrze dla danych z więcej niż dwiema klasami, co ma miejsce w przypadku zestawu danych tęczówki.
from sklearn.decomposition import PCA
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
# load data
iris = load_iris()
# initiate PCA and classifier
pca = PCA()
classifier = DecisionTreeClassifier()
# transform / fit
X_transformed = pca.fit_transform(iris.data)
classifier.fit(X_transformed, iris.target)
# predict "new" data
# (I'm faking it here by using the original data)
newdata = iris.data
# transform new data using already fitted pca
# (don't re-fit the pca)
newdata_transformed = pca.transform(newdata)
# predict labels using the trained classifier
pred_labels = classifier.predict(newdata_transformed)
SciKit learn ma wygodne narzędzie o nazwie Pipeline, które pozwala łączyć ze sobą transformatory i końcowy klasyfikator:
# you can make this a lot easier using Pipeline
from sklearn.pipeline import Pipeline
# fits PCA, transforms data and fits the decision tree classifier
# on the transformed data
pipe = Pipeline([('pca', PCA()),
('tree', DecisionTreeClassifier())])
pipe.fit(iris.data, iris.target)
pipe.predict(newdata)
Jest to szczególnie przydatne podczas sprawdzania poprawności krzyżowej, ponieważ zapobiega przypadkowemu dopasowaniu ŻADNEGO kroku potoku w zestawie danych testowania:
from sklearn.cross_validation import cross_val_score
print cross_val_score(pipe, iris.data, iris.target)
# [ 0.96078431 0.90196078 1. ]
Nawiasem mówiąc, może nie być konieczne użycie PCA, aby uzyskać dobre wyniki klasyfikacji. Zestaw danych tęczówki nie ma wielu wymiarów, a drzewa decyzyjne będą już dobrze działać na nietransformowanych danych.