paradiseo/edo/src/edoSamplerBinomial.h

66 lines
1.8 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>
*/
#ifndef _edoSamplerBinomial_h
#define _edoSamplerBinomial_h
#include <utils/eoRNG.h>
#include "edoSampler.h"
#include "edoBinomial.h"
/** A sampler for an edoBinomial distribution.
*
* @ingroup Samplers
* @ingroup Binomial
*/
template< class EOT, class D = edoBinomial<EOT> >
class edoSamplerBinomial : public edoSampler<D>
{
public:
edoSamplerBinomial() : edoSampler<D>() {}
EOT sample( D& distrib )
{
unsigned int size = distrib.size();
assert(size > 0);
// The point we want to draw
// x = {x1, x2, ..., xn}
EOT solution;
// Sampling all dimensions
for( unsigned int i = 0; i < size; ++i ) {
// Toss a coin, biased by the proba of being 1.
solution.push_back( rng.flip(distrib[i]) );
}
return solution;
}
};
#endif // !_edoSamplerBinomial_h