Keras 中的多任务学习

Question

我有两个不同的数据集，我想尝试多任务学习。我的问题是我能找到的所有示例都有两个不同的训练输入，但标签是相同的。我的问题是：我可以有不同的标签吗？这是我现在的代码：

input1 = Sequential()
input1.add(Embedding(vocabulary_size, embedding_size, 
input_length=longest_sen_input1))
input1.add(Bidirectional(LSTM(units=embedding_size)))
input1.add(Dense(len(document), activation='softmax'))

input2 = Sequential()
input2.add(Embedding(vocabulary_size, embedding_size, 
input_length=longest_sen_input2))
input2.add(Bidirectional(LSTM(units=embedding_size)))
input2.add(Dense(len(document), activation='softmax'))

model = Sequential()
model.add(Merge([input1, input2], mode='sum'))
model.add(Dense(len(document), activation='softmax'))
model.compile(loss='categorical_crossentropy',optimizer='adam')

model.fit([X_train_input1, X_train_input2], [Y_train_input1, Y_train_input2], epochs=100)

我尝试插入 [Y_train_input1, Y_train_input2]，但出现此错误：

Error when checking model target: the list of Numpy arrays that you are passing to your model is not the size the model expected. Expected to see 1 array(s), but instead got the following list of 2 arrays: [array([[0., 0., 0., ..., 0., 0., 0.],
   [0., 0., 0., ..., 0., 0., 0.],
   [0., 0., 0., ..., 0., 0., 0.],
   ...,
   [0., 0., 0., ..., 1., 0., 0.],
   [0., 0., 0., ..., 0., 0., 0....

有人知道如何使用两个输入（X_train_input1/Y_train_input1 和 X_train_input2/Y_train_input2）执行多任务学习，返回一个共同的预测？

编辑 我的模型现在似乎可以工作了，我只是改变了

model.fit([X_train_input1, X_train_input2], [Y_train_input1, Y_train_input2], epochs=100)

在

model.fit([X_train_input1, X_train_input2], Y_train, epochs=100)

然后我尝试像这样测试模型

multitask_model.predict_classes(X_test)

我有这个错误：

ValueError: Error when checking model : the list of Numpy arrays that you are passing to your model is not the size the model expected. Expected to see 2 array(s), but instead got the following list of 1 arrays: [array([[  0,   0,   0, ...,  13,   8, 134],
   [  0,   0,   0, ...,  33,  87,  19],
   [  0,   0,   0, ...,  27,   1,   4],
   ...,
   [  0,   0,   0, ...,   1,  10,   8],
   [  0...

我错过了什么？

Answer 1

我用这个y_pred = model.predict([X_test,X_test]).argmax(axis=1)解决了同样的问题 你必须传递两个数组而不是一个

Answer 2

您的模型只有一个输出，而您传递了两个：Y_train_input1、Y_train_input2。

如果您的目标不是合并两个模型，那么您应该将它们分开。当您合并/求和输出时，您最终只有一个输出。

您是否真的打算拥有两个不同的独立模型，而它们之间没有任何交互？

• 要么你有一个共同的输出和一个共同的Y_train，要么
• 您有两个单独的输出和两个不同的目标。