具有多个输出的KerasClassifier

时间:2019-09-06 11:39:00

标签: python keras scikit-learn

我有带多个输出的biLSTM神经网络。每个输出都是大小不同的向量。

我用KerasClassifier包裹了网络,以便能够将其与scikit-learn一起使用(例如,用于网格搜索)。不幸的是,它在尝试适合时失败了:

ValueError: could not broadcast input array from shape (100,10,3) into shape (100,10)

以下是复制它的最小示例:

from keras.wrappers.scikit_learn import KerasClassifier
import numpy as np

def create_model():
    return None

clf = KerasClassifier(build_fn=create_model)

samples_count = 100
sample_size = 10
outputs_sizes = [3,4,1]

y = [np.zeros((samples_count, sample_size, output_size)) 
     for output_size 
     in outputs_sizes]

X = np.zeros((samples_count, 100))

clf.fit(X, y)

如何将KerasClassifier与多个输出配合使用?

0 个答案:

没有答案