Function mlp_model

examples/rl_deep_q_learning.py:11–21

(n_actions, batch_size=64)

def mlp_model(n_actions, batch_size=64):
    model = NeuralNet(
        layers=[Dense(32), Activation("relu"), Dense(n_actions)],
        loss="mse",
        optimizer=Adam(),
        metric="mse",
        batch_size=batch_size,
        max_epochs=1,
        verbose=False,
    )
    return model


model = DQN(n_episodes=2500, batch_size=64)