PCA è uno strumento di riduzione dimensionale, non un classificatore. In Scikit-Learn, tutti i classificatori e gli stimatori hanno un predict
metodo che PCA non ha . È necessario adattare un classificatore ai dati trasformati PCA. Scikit-Learn ha molti classificatori. Ecco un esempio dell'uso di un albero decisionale su dati trasformati in PCA. Ho scelto il classificatore dell'albero decisionale in quanto funziona bene per i dati con più di due classi, come nel caso del set di dati dell'iride.
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 ha un comodo strumento chiamato Pipeline che ti consente di unire i trasformatori e un classificatore finale:
# 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)
Ciò è particolarmente utile quando si esegue la convalida incrociata in quanto impedisce di ricollegare accidentalmente QUALSIASI passaggio della pipeline sul set di dati di test:
from sklearn.cross_validation import cross_val_score
print cross_val_score(pipe, iris.data, iris.target)
# [ 0.96078431 0.90196078 1. ]
A proposito, potrebbe non essere nemmeno necessario utilizzare PCA per ottenere buoni risultati di classificazione. Il set di dati dell'iride non ha molte dimensioni e gli alberi delle decisioni funzioneranno già bene sui dati non trasformati.