PCA ist ein Werkzeug zur Dimensionsreduzierung, kein Klassifikator. In Scikit-Learn haben alle Klassifikatoren und Schätzer eine predict
Methode, die PCA nicht verwendet . Sie müssen einen Klassifikator an die PCA-transformierten Daten anpassen. Scikit-Learn hat viele Klassifikatoren. Hier ist ein Beispiel für die Verwendung eines Entscheidungsbaums für PCA-transformierte Daten. Ich habe den Entscheidungsbaumklassifikator gewählt, da er für Daten mit mehr als zwei Klassen gut funktioniert, was beim Iris-Dataset der Fall ist.
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 verfügt über ein praktisches Tool namens Pipeline, mit dem Sie Transformatoren und einen endgültigen Klassifikator miteinander verketten können :
# 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)
Dies ist besonders nützlich, wenn Sie eine Kreuzvalidierung durchführen, da Sie nicht versehentlich einen Schritt der Pipeline in Ihren Testdatensatz einpassen:
from sklearn.cross_validation import cross_val_score
print cross_val_score(pipe, iris.data, iris.target)
# [ 0.96078431 0.90196078 1. ]
Übrigens müssen Sie möglicherweise nicht einmal PCA verwenden, um gute Klassifizierungsergebnisse zu erzielen. Der Iris-Datensatz hat nicht viele Dimensionen und Entscheidungsbäume funktionieren bereits gut mit den nicht transformierten Daten.