2017-01-05 76 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