前提・実現したいこと
Python3.5でkerasと機械学習用データcifar10を用いてCNNを実装している途中です。
学習段階でエラーが出ていますが、その対処法と根本的な原因がわかりません。
発生している問題・エラーメッセージ
Error when checking target: expected dense_37 to have shape (10,) but got array with shape (1,)
該当のソースコード
自分が実装したコードです。
#インポート import tensorflow as tf import keras import numpy as np import matplotlib.pyplot as plt #データの取得 from keras.datasets import cifar10 (x_train, y_train), (x_test, y_test) = cifar10.load_data() #画像(行列)の正規化 x_train = x_train / 255.0 x_test = x_test / 255.0 #CNNモデルを構築 def CNN_model(): model = keras.models.Sequential() #1st convolutional layer model.add(keras.layers.Conv2D(96, 11, strides=(4, 4), padding='valid', data_format='channels_last', activation='relu', bias_initializer='ones', input_shape=(32,32,3))) model.add(keras.layers.normalization.BatchNormalization(axis=1)) model.add(keras.layers.MaxPooling2D(pool_size=(3, 3), strides=(2, 2), padding='valid', data_format='channels_last')) #2nd convolutional layer model.add(keras.layers.Conv2D(256, 2, strides=(4, 4), padding='valid', data_format='channels_last', activation='relu', bias_initializer='zeros')) model.add(keras.layers.normalization.BatchNormalization(axis=1)) model.add(keras.layers.MaxPooling2D(pool_size=(1, 1), strides=(2, 2), padding='valid', data_format='channels_last')) #3rd flatten and whole junction layer model.add(keras.layers.Flatten()) model.add(keras.layers.Dense(4096)) model.add(keras.layers.Dropout(0.5)) model.add(keras.layers.Dense(4096)) model.add(keras.layers.Dropout(0.5)) #4th Softmax layer model.add(keras.layers.Dense(10, activation='softmax')) #コンパイル model.compile(optimizer=keras.optimizers.SGD(lr=0.01), loss='categorical_crossentropy', metrics=['accuracy']) return model
#CNNモデルのサマリー model = CNN_model() model.summary()
#学習 early_stopping = keras.callbacks.EarlyStopping(monitor='val_loss', patience=3, verbose=1, mode='auto') history = model.fit(x_train, y_train, epochs=15, batch_size=128, shuffle=True, validation_split=0.25, callbacks=[early_stopping])
エラーメッセージ
ValueError: Error when checking target: expected dense_46 to have shape (10,) but got array with shape (1,)
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/04/05 10:43