问题描述
我正在按照以下代码创建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()
- 我想知道GRU层在训练时是否不会处理被屏蔽的输入。有什么办法可以检查吗?
- 如果1为true,那么评估和预测会如何继续使用该遮罩?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)