|
mlpack
|
#include <mlpack/core.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/ann/loss_functions/sigmoid_cross_entropy_error.hpp>#include <mlpack/methods/reinforcement_learning/async_learning.hpp>#include <mlpack/methods/reinforcement_learning/environment/cart_pole.hpp>#include <mlpack/methods/reinforcement_learning/policy/greedy_policy.hpp>#include <mlpack/methods/reinforcement_learning/policy/aggregated_policy.hpp>#include <mlpack/methods/reinforcement_learning/training_config.hpp>#include <ensmallen.hpp>#include "catch.hpp"
Functions | |
| TEST_CASE ("OneStepQLearningTest", "[AsyncLearningTest]") | |
| TEST_CASE ("OneStepSarsaTest", "[AsyncLearningTest]") | |
| TEST_CASE ("NStepQLearningTest", "[AsyncLearningTest]") | |
Test for async deep RL methods.
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.
| TEST_CASE | ( | "OneStepQLearningTest" | , |
| "" | [AsyncLearningTest] | ||
| ) |
This is for the Travis CI server, in your own machine you should use more threads.
| TEST_CASE | ( | "OneStepSarsaTest" | , |
| "" | [AsyncLearningTest] | ||
| ) |
This is for the Travis CI server, in your own machine you shuold use more threads.
| TEST_CASE | ( | "NStepQLearningTest" | , |
| "" | [AsyncLearningTest] | ||
| ) |
This is for the Travis CI server, in your own machine you shuold use more threads.
1.8.13