在sklearn Pipeline中包含自定义特征提取方法

时间:2017-02-14 07:04:46

标签: python machine-learning scikit-learn feature-extraction

在sklearn中,可以通过以下方式定义管道:

from sklearn.pipeline import Pipeline
from sklearn.svm import SVC
from sklearn.decomposition import PCA
estimators = [('reduce_dim', PCA()), ('clf', SVC())]
pipe = Pipeline(estimators)

是否也可以包含自定义特征提取方法,如

extract_features(image, cspace='RGB', 
                        pix_per_cell=128, cell_per_block=32, 
                        hog_channel=10)

我该怎么做?

0 个答案:

没有答案