reward_clipping_test.cpp File Reference

#include <mlpack/core.hpp>
#include <mlpack/methods/reinforcement_learning/environment/mountain_car.hpp>
#include <mlpack/methods/reinforcement_learning/q_networks/simple_dqn.hpp>
#include <mlpack/methods/reinforcement_learning/environment/continuous_mountain_car.hpp>
#include <mlpack/methods/reinforcement_learning/environment/cart_pole.hpp>
#include <mlpack/methods/reinforcement_learning/environment/acrobot.hpp>
#include <mlpack/methods/reinforcement_learning/environment/pendulum.hpp>
#include <mlpack/methods/reinforcement_learning/environment/reward_clipping.hpp>
#include <mlpack/methods/ann/ffn.hpp>
#include <mlpack/methods/ann/init_rules/gaussian_init.hpp>
#include <mlpack/methods/ann/layer/layer.hpp>
#include <mlpack/methods/ann/loss_functions/mean_squared_error.hpp>
#include <mlpack/methods/reinforcement_learning/q_learning.hpp>
#include <mlpack/methods/reinforcement_learning/policy/greedy_policy.hpp>
#include <mlpack/methods/reinforcement_learning/training_config.hpp>
#include <ensmallen.hpp>
#include "catch.hpp"

Include dependency graph for reward_clipping_test.cpp:

Functions
	TEST_CASE ("ClippedRewardTest", "[RewardClippingTest]")
	TEST_CASE ("RewardClippedAcrobotWithDQN", "[RewardClippingTest]")
	Test DQN in Acrobot task. More...

Detailed Description

Author

Shashank Shekhar

Test for the reward clipping wrapper for reinforcement learning environments.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.

Function Documentation

TEST_CASE()

TEST_CASE	(	"RewardClippedAcrobotWithDQN"	,
		""	[RewardClippingTest]
	)

Test DQN in Acrobot task.

I am using a thresold of -380 to check convegence.