2
我有一個問題,我想測試多個不具有相同命名參數的模型。您如何使用randomizedSearchCV
中管道的參數列表,就像在本例中可以使用GridSearchCV
一樣?sklearn randomizedSearchCV中的參數列表像GridSearchCV?
實施例從:
http://scikit-learn.org/stable/auto_examples/plot_compare_reduction.html
N_FEATURES_OPTIONS = [2, 4, 8]
C_OPTIONS = [1, 10, 100, 1000]
param_grid = [
{
'reduce_dim': [PCA(iterated_power=7), NMF()],
'reduce_dim__n_components': N_FEATURES_OPTIONS,
'classify__C': C_OPTIONS
},
{
'reduce_dim': [SelectKBest(chi2)],
'reduce_dim__k': N_FEATURES_OPTIONS,
'classify__C': C_OPTIONS
},
]
grid = GridSearchCV(pipe, cv=3, n_jobs=2, param_grid=param_grid)
digits = load_digits()
grid.fit(digits.data, digits.target)