PCA - это инструмент уменьшения размеров, а не классификатор. В Scikit-Learn у всех классификаторов и оценщиков есть predict
метод, которого нет у PCA . Вам необходимо установить классификатор на данные, преобразованные PCA. Scikit-Learn имеет много классификаторов. Вот пример использования дерева решений для PCA-преобразованных данных. Я выбрал классификатор дерева решений, поскольку он хорошо работает для данных с более чем двумя классами, как в случае с набором данных iris.
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 имеет удобный инструмент под названием Pipeline, который позволяет объединить в цепочку трансформаторы и окончательный классификатор:
# 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)
Это особенно полезно при перекрестной проверке, поскольку предотвращает случайную повторную установку ЛЮБОГО шага конвейера в наборе данных тестирования:
from sklearn.cross_validation import cross_val_score
print cross_val_score(pipe, iris.data, iris.target)
# [ 0.96078431 0.90196078 1. ]
Кстати, вам может даже не понадобиться использовать PCA, чтобы получить хорошие результаты классификации. Набор данных iris не имеет большого количества измерений, и деревья решений уже будут хорошо работать с нетрансформированными данными.