cne.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_OPTIMIZERS_CNE_CNE_HPP
#define MLPACK_CORE_OPTIMIZERS_CNE_CNE_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace optimization {

class CNE
{
 public:
  CNE(const size_t populationSize = 500,
      const size_t maxGenerations = 5000,
      const double mutationProb = 0.1,
      const double mutationSize = 0.02,
      const double selectPercent = 0.2,
      const double tolerance = 1e-5,
      const double objectiveChange = 1e-5);

  template<typename DecomposableFunctionType>
  double Optimize(DecomposableFunctionType& function, arma::mat& iterate);

  size_t PopulationSize() const { return populationSize; }
  size_t& PopulationSize() { return populationSize; }

  size_t MaxGenerations() const { return maxGenerations; }
  size_t& MaxGenerations() { return maxGenerations; }

  double MutationProbability() const { return mutationProb; }
  double& MutationProbability() { return mutationProb; }

  double MutationSize() const { return mutationSize; }
  double& MutationSize() { return mutationSize; }

  double SelectionPercentage() const { return selectPercent; }
  double& SelectionPercentage() { return selectPercent; }

  double Tolerance() const { return tolerance; }
  double& Tolerance() { return tolerance; }

  double ObjectiveChange() const { return objectiveChange; }
  double& ObjectiveChange() { return objectiveChange; }

 private:
  void Reproduce();

  void Mutate();

  void Crossover(const size_t mom,
                 const size_t dad,
                 const size_t dropout1,
                 const size_t dropout2);

  arma::cube population;

  arma::vec fitnessValues;

  arma::uvec index;

  size_t populationSize;

  size_t maxGenerations;

  double mutationProb;

  double mutationSize;

  double selectPercent;

  double tolerance;

  double objectiveChange;

  size_t numElite;

  size_t elements;
};

} // namespace optimization
} // namespace mlpack

// Include implementation.
#include "cne_impl.hpp"

#endif