微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Keras GRU层可以与Masking层一起使用吗?

如何解决Keras GRU层可以与Masking层一起使用吗?

我正在按照以下代码创建Keras GRU模型

    inputs = Input(batch_shape=(args.batch_size,1,args.train_n_items + 1))
    masks = Masking(mask_value=0.0)(inputs)
    gru,gru_states = GRU(hidden_units,stateful=True,return_state=True,name="GRU")(masks)
    drop2 = Dropout(0.25)(gru)
    predictions = Dense(args.train_n_items + 1,activation='softmax')(drop2)
    model = Model(inputs=inputs,outputs=[predictions])
    opt = keras.optimizers.Adam(lr=0.001,beta_1=0.9,beta_2=0.999,epsilon=None,decay=0.0,amsgrad=False)
    recall_metric = keras.metrics.Recall(top_k=10)
    model.compile(loss=categorical_crossentropy,optimizer=opt,metrics=[recall_metric])
    model.summary()
  1. 我想知道GRU层在训练时是否不会处理被屏蔽的输入。有什么办法可以检查吗?
  2. 如果1为true,那么评估和预测会如何继续使用该遮罩?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。