簡體   English   中英

keras模型中的fit_generator()

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM