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.