孔橙 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去掉就可以了

    评论

报告相同问题?

悬赏问题

  • ¥20 蓝牙耳机怎么查看日志
  • ¥15 R语言 拟时序分析降维图如何减少分支
  • ¥15 Fluent齿轮搅油
  • ¥15 八爪鱼爬数据为什么自己停了
  • ¥15 交替优化波束形成和ris反射角使保密速率最大化
  • ¥15 树莓派与pix飞控通信
  • ¥15 自动转发微信群信息到另外一个微信群
  • ¥15 outlook无法配置成功
  • ¥30 这是哪个作者做的宝宝起名网站
  • ¥60 版本过低apk如何修改可以兼容新的安卓系统