小编hak*_*rus的帖子

在Keras中使用Tensorflow Huber损失

我试图在keras模型中使用huber loss(编写DQN),但是我的结果很糟糕,我认为我做错了.我的代码如下.

model = Sequential()
model.add(Dense(output_dim=64, activation='relu', input_dim=state_dim))
model.add(Dense(output_dim=number_of_actions, activation='linear'))
loss = tf.losses.huber_loss(delta=1.0)
model.compile(loss=loss, opt='sgd')
return model
Run Code Online (Sandbox Code Playgroud)

python reinforcement-learning keras tensorflow

11
推荐指数
2
解决办法
9070
查看次数