104 lines
3.1 KiB
C++
104 lines
3.1 KiB
C++
/*
|
|
The Evolving Distribution Objects framework (EDO) is a template-based,
|
|
ANSI-C++ evolutionary computation library which helps you to write your
|
|
own estimation of distribution algorithms.
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
Copyright (C) 2013 Thales group
|
|
*/
|
|
/*
|
|
Authors:
|
|
Johann Dréo <johann.dreo@thalesgroup.com>
|
|
*/
|
|
|
|
#include <vector>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <cmath>
|
|
|
|
#include <eo>
|
|
#include <edo>
|
|
#include <ga.h> // for Bools
|
|
|
|
template<class T>
|
|
void print( T v, std::string sep=" " )
|
|
{
|
|
for( typename T::iterator it=v.begin(); it!=v.end(); ++it ) {
|
|
std::cout << *it << sep;
|
|
}
|
|
std::cout << std::endl;
|
|
}
|
|
|
|
int main(int ac, char** av)
|
|
{
|
|
typedef eoBit<double> Bools;
|
|
|
|
eoParser parser(ac, av);
|
|
|
|
std::string section("Algorithm parameters");
|
|
unsigned int popsize = parser.createParam((unsigned int)100000, "popSize", "Population Size", 'P', section).value(); // P
|
|
unsigned int dim = parser.createParam((unsigned int)20, "dim", "Dimension size", 'd', section).value(); // d
|
|
double proba = parser.createParam((double)0.5, "proba", "Probability to estimate", 'b', section).value(); // b
|
|
|
|
if( parser.userNeedsHelp() ) {
|
|
parser.printHelp(std::cout);
|
|
exit(1);
|
|
}
|
|
|
|
make_help(parser);
|
|
|
|
|
|
// This generate a random boolean when called...
|
|
eoBooleanGenerator flip( proba );
|
|
|
|
// ... by this initializator...
|
|
eoInitFixedLength<Bools> init( dim, flip );
|
|
|
|
// ... when building this pop.
|
|
std::cout << "Init a pop of eoBit of size " << popsize << " at random with p(x=1)=" << proba << std::endl;
|
|
eoPop<Bools> pop( popsize, init );
|
|
// print( pop, "\n" );
|
|
|
|
// EDO triplet
|
|
edoBinomial<Bools> distrib(dim,0.0);
|
|
edoEstimatorBinomial<Bools> estimate;
|
|
edoSamplerBinomial<Bools> sample;
|
|
|
|
std::cout << "Estimate a distribution from the init pop" << std::endl;
|
|
distrib = estimate(pop);
|
|
print( distrib );
|
|
// std::cout << std::endl;
|
|
|
|
std::cout << "Sample a new pop from the init distrib" << std::endl;
|
|
pop.clear();
|
|
for( unsigned int i=0; i<popsize; ++i ) {
|
|
pop.push_back( sample( distrib ) );
|
|
}
|
|
// print( pop, "\n" );
|
|
|
|
std::cout << "Estimate a distribution from the sampled pop" << std::endl;
|
|
distrib = estimate(pop);
|
|
print( distrib );
|
|
// std::cout << std::endl;
|
|
|
|
double sum=0;
|
|
for( unsigned int i=0; i<distrib.size(); ++i ) {
|
|
sum += distrib[i];
|
|
}
|
|
double e = sum / dim;
|
|
|
|
std::cout << "Estimated initial proba = " << e << std::endl;
|
|
}
|