2017-01-05 120 views
2

我正在尝试为使用keras构建的神经网络执行参数调整。这是我的代码上导致错误的行注释:与Keras神经网络的GridSearch

from sklearn.cross_validation import StratifiedKFold, cross_val_score 
from sklearn import grid_search 
from sklearn.metrics import classification_report 
import multiprocessing 

from keras.models import Sequential 
from keras.layers import Dense 
from sklearn.preprocessing import LabelEncoder 
from keras.utils import np_utils 
from keras.wrappers.scikit_learn import KerasClassifier 
import numpy as np 


def tuning(X_train,Y_train,X_test,Y_test): 

    in_size=X_train.shape[1] 
    num_cores=multiprocessing.cpu_count() 
    model = Sequential() 
    model.add(Dense(in_size, input_dim=in_size, init='uniform', activation='relu')) 
    model.add(Dense(8, init='uniform', activation='relu')) 
    model.add(Dense(1, init='uniform', activation='sigmoid')) 
    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) 

    batch_size = [10, 20, 40, 60, 80, 100] 
    epochs = [10,20] 
    param_grid = dict(batch_size=batch_size, nb_epoch=epochs) 

    k_model = KerasClassifier(build_fn=model, verbose=0) 
    clf = grid_search.GridSearchCV(estimator=k_model, param_grid=param_grid, cv=StratifiedKFold(Y_train, n_folds=10, shuffle=True, random_state=1234), 
        scoring="accuracy", verbose=100, n_jobs=num_cores) 

    clf.fit(X_train, Y_train) #ERROR HERE 

    print("Best parameters set found on development set:") 
    print() 
    print(clf.best_params_) 
    print() 
    print("Grid scores on development set:") 
    print() 
    for params, mean_score, scores in clf.grid_scores_: 
    print("%0.3f (+/-%0.03f) for %r" 
     % (mean_score, scores.std() * 2, params)) 
    print() 
    print("Detailed classification report:") 
    print() 
    print("The model is trained on the full development set.") 
    print("The scores are computed on the full evaluation set.") 
    print() 
    y_true, y_pred = Y_test, clf.predict(X_test) 
    print(classification_report(y_true, y_pred)) 
    print() 

这是错误报告:

clf.fit(X_train, Y_train) 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/grid_search.py", line 804, in fit 
    return self._fit(X, y, ParameterGrid(self.param_grid)) 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/grid_search.py", line 553, in _fit 
    for parameters in parameter_iterable 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/externals/joblib/parallel.py", line 800, in __call__ 
    while self.dispatch_one_batch(iterator): 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/externals/joblib/parallel.py", line 658, in dispatch_one_batch 
    self._dispatch(tasks) 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/externals/joblib/parallel.py", line 566, in _dispatch 
    job = ImmediateComputeBatch(batch) 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/externals/joblib/parallel.py", line 180, in __init__ 
    self.results = batch() 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/externals/joblib/parallel.py", line 72, in __call__ 
    return [func(*args, **kwargs) for func, args, kwargs in self.items] 
    File "/usr/local/lib/python2.7/dist-packages/sklearn/cross_validation.py", line 1531, in _fit_and_score 
    estimator.fit(X_train, y_train, **fit_params) 
    File "/usr/local/lib/python2.7/dist-packages/keras/wrappers/scikit_learn.py", line 135, in fit 
    **self.filter_sk_params(self.build_fn.__call__)) 
TypeError: __call__() takes at least 2 arguments (1 given) 

我缺少的东西?网格搜索顺利随机森林,svm和逻辑回归。我只有神经网络的问题。

回答

2

这里的错误表明build_fn需要有2个参数,如param_grid的#参数所示。

所以,你需要明确地定义一个新的功能和使用,作为build_fn=make_model

def make_model(batch_size, nb_epoch): 
    model = Sequential() 
    model.add(Dense(in_size, input_dim=in_size, init='uniform', activation='relu')) 
    model.add(Dense(8, init='uniform', activation='relu')) 
    model.add(Dense(1, init='uniform', activation='sigmoid')) 
    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) 
    return model 

还要检查keras/examples/mnist_sklearn_wrapper.py其中GridSearchCV用于超参数的搜索。

+0

你是对的!我无法弄清楚。谢谢! –

+0

没问题。此外,现在我注意到您正在'batch_size'&'nb_epoch'上进行参数搜索。这些是keras模型拟合参数。在目前的结构中,它不会工作。虽然我认为可以做到这一点,但需要更改包装代码。相反,我会建议你使用https://github.com/maxpumperla/hyperas,你可以很容易地使用它们作为超参数 – indraforyou

0

我想你也许使用scikit-learn 0.16或更早版本。
昨天刚刚遇到同样的问题,经过一些变通后,我才知道将scikit-learn从0.16升级到0.18可以解决问题。

clf.fit(X_train, Y_train) #SHOULD WORK with scikit-learn 0.18 

一两件事,从0.16 0.18不同的是在GridSearchCV不拿出sklearn.grid_searchsklearn.model_selection