與keras`fit_generator（）外形問題`我有與numpy的陣列的形狀，一個看似簡單的問題

Question

從我使用keras的fit_generator函數中的自定義生成函數返回。

發電機功能類似於此：

def data_generator(full_data, encoder): 
    for s in full_data: 
     in1_X = encoder.encode(s[:,0]) 
     in2_X = encoder.encode(s[:,1]) 
     out1_y = encoder.encode(s[:,2]) 
     out2_y = encoder.encode(s[:,3]) 
     X = [in1_X, in2_X] 
     y = [out1_y, out2_y] 
     yield (X,y) 

我可以通過使用for循環和打印的形狀，這只是返回(60,)

然而，在調用這個時候得到in1_X返回的形狀使用fit_generator()功能，它失敗：

train_data_gen = data_generator(full_data, encoder) 


main_in = Input(shape=(seq_len,), name='main_input') 

# ... 
# define model 
# ... 

joint_model.fit_generator(train_data_gen, steps_per_epoch=2000, epochs=2) 

從其中輸出是這樣的：

Error when checking input: 
expected main_input to have shape (None, 60) but got array with shape (60, 1) 

我怎樣才能得到這個不改變形狀(60,) numpy陣列形狀(60, 1)？其他人遇到過這個問題嗎？

Answer 1

我們發現對這個問題的迴應。請在下面的PR檢查的解釋和評論：https://github.com/fchollet/keras/issues/4641 提供一個二維數組解決了問題我們。