ディープラーニング にて、グリッドサーチを使用したいのですが、
以下のコードの最後の行(grid_result = grid.fit(x_train, y_train))
にて、以下のようなエラーが発生します。
どこがおかしいかご教授いただければ嬉しいです。
何卒宜しくお願い致します。
エラー
ValueError: dense is not a legal parameter
コード
from keras.models import Model from keras.applications.resnet50 import ResNet50 from keras.layers import Dense,Input,Flatten,Dropout,BatchNormalization def create_normal_model(): input_tensor = Input(shape=(IMAGE_SIZE, IMAGE_SIZE, 3)) base_model = ResNet50(weights='imagenet', include_top=False,input_tensor=input_tensor) for layer in base_model.layers: layer.trainable=False#ResNetの重みは固定 x = base_model.output x=Dropout(dropout)(x) x=Flatten()(x) x = Dense(dense, activation='relu')(x) x=BatchNormalization()(x) x=Dropout(dropout)(x) predictions = Dense(N_CATEGORIES, activation='softmax')(x) cn_model = Model(inputs=base_model.input, outputs=predictions) cn_model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) return cn_model batch_size = [16, 32, 64] epochs = [10, 25, 50] dropout = [0.1, 0.3, 0.5] dense = [256, 1024, 4092] N_CATEGORIES = 2 IMAGE_SIZE = 300 from keras.wrappers.scikit_learn import KerasClassifier from sklearn.model_selection import GridSearchCV model_grid = KerasClassifier(build_fn=create_normal_model, verbose=0) param_grid = dict(batch_size=batch_size, epochs=epochs, dropout=dropout, dense=dense) grid = GridSearchCV(estimator=model_grid, param_grid=param_grid) grid_result = grid.fit(x_train, y_train)
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。