1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
|
/**
* @file cmaes_test.cpp
* @author Marcus Edel
* @author Kartik Nighania
* @author Conrad Sanderson
*
* ensmallen 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 ensmallen. If not, see
* http://www.opensource.org/licenses/BSD-3-Clause for more information.
*/
#if defined(ENS_USE_COOT)
#include <armadillo>
#include <bandicoot>
#endif
#include <ensmallen.hpp>
#include "catch.hpp"
#include "test_function_tools.hpp"
#include "test_types.hpp"
using namespace ens;
using namespace ens::test;
/**
* Run CMA-ES with the full selection policy on logistic regression and
* make sure the results are acceptable.
*/
TEMPLATE_TEST_CASE("CMAES_LogisticRegressionFunction", "[CMAES]",
ENS_FULLPREC_TEST_TYPES)
{
BoundaryBoxConstraint<TestType> b(-10, 10);
CMAES<FullSelection, BoundaryBoxConstraint<TestType>> cmaes(
0, b, 32, 500, 1e-3);
cmaes.StepSize() = 0.6;
LogisticRegressionFunctionTest<TestType>(cmaes,
Tolerances<TestType>::LRTrainAcc, Tolerances<TestType>::LRTestAcc, 5);
}
/**
* Run CMA-ES with the random selection policy on logistic regression and
* make sure the results are acceptable.
*/
TEMPLATE_TEST_CASE("ApproxCMAES_LogisticRegressionFunction", "[CMAES]",
ENS_TEST_TYPES)
{
BoundaryBoxConstraint<TestType> b(-10, 10);
ApproxCMAES<BoundaryBoxConstraint<TestType>> cmaes(256, b, 16, 500, 1e-3);
cmaes.StepSize() = 0.6;
LogisticRegressionFunctionTest<TestType>(cmaes,
Tolerances<TestType>::LRTrainAcc, Tolerances<TestType>::LRTestAcc, 5);
}
/**
* Run CMA-ES with the random selection and empty transformation policies
* on logistic regression and make sure the results are acceptable.
* Use arma::fmat.
*/
TEMPLATE_TEST_CASE("ApproxCMAES_EmptyTransformationLogisticRegressionFunction",
"[CMAES]", ENS_TEST_TYPES)
{
ApproxCMAES<EmptyTransformation<TestType>>
cmaes(0, EmptyTransformation<TestType>(), 16, 500, 1e-3);
LogisticRegressionFunctionTest<TestType>(cmaes, 0.01, 0.02, 5);
}
|