首页 > 解决方案 > 使用 Keras 进行层级联

问题描述

我打算有以下设计:


在此处输入图像描述


但是我的代码似乎不起作用:

import numpy as np
from keras.models import Model
from keras.layers import Dense, Input, Concatenate
from keras import optimizers

trainX1 = np.array([[1,2],[3,4],[5,6],[7,8]])   # fake training data
trainY1 = np.array([[1],[2],[3],[4]])           # fake label

trainX2 = np.array([[2,3],[4,5],[6,7]])
trainY2 = np.array([[1],[2],[3]])

trainX3 = np.array([[0,1],[2,3]])
trainY3 = np.array([[1],[2]])

numFeatures = 2
trainXList = [trainX1, trainX2, trainX3]
trainYStack = np.vstack((trainY1,trainY2,trainY3))
inputList = []
modelList = []
for i,_ in enumerate(trainXList):
    tempInput= Input(shape = (numFeatures,))
    m = Dense(10, activation='tanh')(tempInput)
    inputList.append(tempInput)
    modelList.append(m)

mAll = Concatenate()(modelList)
out = Dense(1, activation='tanh')(mAll)
model = Model(inputs=inputList, outputs=out)

rmsp = optimizers.rmsprop(lr=0.00001)
model.compile(optimizer=rmsp,loss='mse', dropout = 0.1)
model.fit(trainXList, trainYStack, epochs = 1, verbose=0)

错误消息说我的输入数据集的形状不同,但是在我填充训练集以使所有 3 个集的样本数 = 4 之后,我仍然收到错误说尺寸不正确。我可以知道如何正确设计这个网络吗?谢谢!


ps 以下是填充前的错误信息:

ValueError: All input arrays (x) should have the same number of samples. Got array shapes: [(4, 2), (3, 2), (2, 2)]

这是填充后的错误消息(发生在代码的最后一行):

ValueError: Input arrays should have the same number of samples as target arrays. Found 4 input samples and 12 target samples.

标签: machine-learningkeras

解决方案


您的输入形状对于给定的输入是错误的。您为输入分配了 numFeatures 的大小,但实际上您有二维数组,它们是不同的 (4,2)(3,2)(2,2)。我不确定你的问题,但样本数量和特征数量似乎是相反的。

tempInput= Input(shape = (numFeatures,))

此外,您的 y 也很奇怪。通常你有 X (number_of samples, num_features) 和 y with (number of samples, labels)。

用于model.summary()查看您的网络的外观。


推荐阅读