Kerasを使って2入力2出力のモデルを作成しています。
学習後、テストデータを使ってpredictを行いたいのですが、引数に2つのnumpy配列を渡す方法が分からなかったため、質問させてください。
以下モデルの定義です。入力は100次元2つ、出力はコサイン類似度とcategorical_crossentropyになります。
fitメソッドで学習は完了しています。
Python
1# データの分割 2test_input_1 = data1[:200] 3test_input_2 = data2[:200] 4test_out_1 = data3[:200] 5test_out_2 = data4[:200] 6 7train_input_1 = data1[200:] 8train_input_2 = data2[200:] 9train_out_1 = data3[200:] 10train_out_2 = data4[200:] 11 12 13# モデルの定義 14input_1 = Input(shape=(100,), name='input_1') 15input_2 = Input(shape=(100,), name='input_2') 16 17x = concatenate([input_1, input_2]) 18middle_layer = Dense(300)(x) 19 20cos_out = Dense(100, activation='sigmoid', 21 name='cos_out')(middle_layer) 22categorical_out = Dense(13, activation='softmax', 23 name='categorical_out')(middle_layer) 24 25model = Model(inputs=[input_1, input_2], 26 outputs=[cos_out, categorical_out]) 27 28model.compile(loss={'cos_out': 'cosine_proximity', 29 'categorical_out': 'categorical_crossentropy'}, 30 optimizer='adam', 31 metrics=['cosine_proximity', 'acc']) 32 33hist = model.fit([train_input_1, train_input_2], 34 [train_out_1, train_out_2], 35 epochs=30)
次にテストですが、
Python
1pred = model.predict([np.array(test_input_1[0]), np.array(test_input_2[0])]) 2print(pred)
として実行したところ、以下のエラーが表示されます。
ValueError: Error when checking input: expected input_1 to have shape (100,) but got array with shape (1,)
入力に100次元必要ですが、ここでは1次元しか得られていない。とのことですが、配列の大きさは学習・テストの入力共に100次元です。
なぜ学習はできてテストができないのか原因を知りたいと思っています。
解決法わかる方いましたら、よろしくお願いいたします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。