SymbRegEvalOp - citanje_svih_funkcija_Tree.cpp

#include <cmath>
#include <ecf/ECF.h>
#include "SymbRegEvalOp.h"
 
 
// called only once, before the evolution – generates training data
bool SymbRegEvalOp::initialize(StateP state)
{
    nSamples = 10;
    double x = -10;
    for(uint i = 0; i < nSamples; i++) {
        domain.push_back(x);
        codomain.push_back(x + sin(x));
        x += 2;
    }
 
    // dohvati prvi genotip - nadamo se da je to Tree
    Tree::Tree* tree = (Tree::Tree*) state->getGenotypes().at(0).get();
 
    // izlistaj sve koristene funkcije
    std::map<std::string, Tree::PrimitiveP>::iterator it = tree->primitiveSet_->mAllPrimitives_.begin();
    while(it != tree->primitiveSet_->mAllPrimitives_.end()) {
        Tree::PrimitiveP func;
        if(func = tree->primitiveSet_->getFunctionByName(it->first)) {
            cout << func->getName();
            cout << "\t" << func->getNumberOfArguments() << endl;
        }
        it++;
    }
 
    // izracunaj rezultat izvodjenja neke funkcije nad konstantama
    // konstante su navedene u parametrima (inace se nece niti pojaviti u stablu)
    Tree::Tree* copyTree = tree->copy();
    std::string izraz = "<Tree size=\"3\">+ D_1 D_2</Tree>";
    XMLNode xTree = XMLNode::parseString(izraz.c_str());
    tree->read(xTree); 
    double result;
    tree->execute(&result);
    cout << result << endl;
 
    return true;
}
 
 
FitnessP SymbRegEvalOp::evaluate(IndividualP individual)
{
    // we try to minimize the function value, so we use FitnessMin fitness (for minimization problems)
    FitnessP fitness (new FitnessMin);
 
    // get the genotype we defined in the configuration file
    Tree::Tree* tree = (Tree::Tree*) individual->getGenotype().get();
    // (you can also use boost smart pointers:)
    //TreeP tree = boost::static_pointer_cast<Tree::Tree> (individual->getGenotype());
 
    double value = 0;
    for(uint i = 0; i < nSamples; i++) {
        // for each test data instance, the x value (domain) must be set
        tree->setTerminalValue("X", &domain[i]);
        // get the y value of the current tree
        double result;
        tree->execute(&result);
        // add the difference
        value += fabs(codomain[i] - result);
    }
    fitness->setValue(value);
 
    return fitness;
}