孔橙 2019-02-15 23:52 采纳率: 0%
浏览 6265

keras模型输出层希望输出的shape是(None,1)但我写的模型输出一个三维的shape(None,10,1)

图片说明

图片说明

### model
taxi_id = Input(shape=(10, 1))
mask_1 = Masking(mask_value=0)(taxi_id)
embedding_1 = Embedding(15000, 14, mask_zero=True)(mask_1)
time_id = Input(shape=(10, 1))
mask_2 = Masking(mask_value=0)(time_id)
embedding_2 = Embedding(7, 4, mask_zero=True)(mask_2)
busy = Input(shape=(10, 1))
mask_3 = Masking(mask_value=0)(busy)
embedding_3 = Embedding(2, 2, mask_zero=True)(mask_3)
concatenate_1 = Concatenate(axis=3)([embedding_1,embedding_2,embedding_3])
concatenate_1 = Lambda(dim_squeeze)(concatenate_1)
num_input = Input(shape=(10, 3))
mask_4 = Masking(mask_value=0, input_shape=())(num_input)
concatenate_2 = Concatenate(axis=2)([concatenate_1, mask_4])
blstm_1 = Bidirectional(LSTM(64,  activation='tanh', return_sequences=True, dropout=0.2, recurrent_dropout=0.2))(concatenate_2)
blstm_2 = Bidirectional(LSTM(128,  activation='tanh', return_sequences=True, dropout=0.2, recurrent_dropout=0.2))(blstm_1)
blstm_3 = Bidirectional(LSTM(64,  activation='tanh', return_sequences=True, dropout=0.2, recurrent_dropout=0.2))(blstm_2)
dense_1 = Dense(128)(blstm_3)
leaky_relu_1 = advanced_activations.LeakyReLU(alpha=0.3)(dense_1)
dense_2 = Dense(128)(leaky_relu_1)
leaky_relu_2 = advanced_activations.LeakyReLU(alpha=0.3)(dense_2)
dense_3 = Dense(128)(leaky_relu_2)
leaky_relu_3 = advanced_activations.LeakyReLU(alpha=0.3)(dense_3)
dense_4 = Dense(128)(leaky_relu_3)
leaky_relu_4 = advanced_activations.LeakyReLU(alpha=0.3)(dense_4)
add_1 = add([leaky_relu_1, leaky_relu_2, leaky_relu_3, leaky_relu_4])
dense_5 = Dense(1)(add_1)
model = Model([taxi_id, time_id, busy, num_input], dense_5)

求教大佬该怎么写能把输出维度降下来

  • 写回答

2条回答 默认 最新

  • 懵啊九 2022-04-08 20:02
    关注

    把最后一个bisltm的return_sequences=True去掉就可以了

    评论

报告相同问题?

悬赏问题

  • ¥100 求三轴之间相互配合画圆以及直线的算法
  • ¥100 c语言,请帮蒟蒻写一个题的范例作参考
  • ¥15 名为“Product”的列已属于此 DataTable
  • ¥15 安卓adb backup备份应用数据失败
  • ¥15 eclipse运行项目时遇到的问题
  • ¥15 关于#c##的问题:最近需要用CAT工具Trados进行一些开发
  • ¥15 南大pa1 小游戏没有界面,并且报了如下错误,尝试过换显卡驱动,但是好像不行
  • ¥15 自己瞎改改,结果现在又运行不了了
  • ¥15 链式存储应该如何解决
  • ¥15 没有证书,nginx怎么反向代理到只能接受https的公网网站