[英]fit_generator() in keras model
def Generate():
i = 0
while 1:
i = i%int(Numb/batch_size)
my_input_batch = my_input[i*batch_size : (i+1)*batch_size]
my_output_batch = my_output[i*batch_size : (i+1)*batch_size]
encoder_input_data = np.array(np.zeros((batch_size, max_encoder_text_length, num_dictonary),dtype='float32'))
decoder_input_data = np.array(np.zeros((batch_size, max_decoder_text_length, num_dictonary),dtype='float32'))
decoder_target_data = np.array(np.zeros((batch_size, max_decoder_text_length, num_dictonary),dtype='float32'))
for i, (text_input, text_output) in enumerate(zip(my_input_batch, my_output_batch)):
for t, word in enumerate(my_input_batch):
encoder_input_data[i, t, word] = 1.
for t, word in enumerate(my_output_batch):
decoder_input_data[i, t, word] = 1.
if t > 0:
decoder_target_data[i, t - 1, word] = 1.
i = i + 1
yield ({encoder_input_data, decoder_input_data}, {decoder_target_data})
我想在fit_generator()
使用fit_generator()
訓練我的模型。 但是,當我開始訓練時,我會出錯:
File "test.py", line 146, in Generate yield ({encoder_input_data, decoder_input_data}, {decoder_target_data})
TypeError: unhashable type: 'numpy.ndarray'
我該如何解決這個問題? 我做錯了什么?
這個特定的語句{encoder_input_data, decoder_input_data}, {decoder_target_data}
是創建一個set
總分numpy
陣列。 該操作是不可能的。
可能是您想將該聲明更改為
yield ((encoder_input_data, decoder_input_data), (decoder_target_data))
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.