aboutsummaryrefslogtreecommitdiff
path: root/DualMLP.cpp
blob: 3a854c0f3cca1fbcad4a4ea50a226ab2de6eaafb (plain)
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
65
66
67
68
69
70
71
72
#include "DualMLP.hpp"

DualMLP::DualMLP(unsigned inputCount, unsigned hiddenNodeCount, unsigned outputCount, NodeClass nodeClass, bool zeroed)
:   m_firstHiddenLayer  (inputCount, hiddenNodeCount, nodeClass, zeroed),
    m_secondHiddenLayer (hiddenNodeCount, hiddenNodeCount, nodeClass, zeroed),
    m_outputLayer       (hiddenNodeCount, outputCount, nodeClass, zeroed)
{
    m_inputCount = inputCount;
    m_hiddenNodeCount = hiddenNodeCount;
    m_outputCount = outputCount;
    m_nodeClass = nodeClass;
}

void DualMLP::setChromosome(const Chromosome &chromosome)
{
    if (chromosome.size() != getChromosomeSize())
    {
        return;
    }

    auto l1 = chromosome.begin() + m_firstHiddenLayer.getChromosomeSize();
    auto l2 = l1 + m_secondHiddenLayer.getChromosomeSize();

    Chromosome chr1(chromosome.begin(), l1);
    Chromosome chr2(l1, l2);
    Chromosome chr3(l2, chromosome.end());

    m_firstHiddenLayer.setChromosome(chr1);
    m_secondHiddenLayer.setChromosome(chr2);
    m_outputLayer.setChromosome(chr3);
}

Chromosome DualMLP::getChromosome() const
{
    Chromosome chromosome;

    Chromosome chr1 = m_firstHiddenLayer.getChromosome();
    Chromosome chr2 = m_secondHiddenLayer.getChromosome();
    Chromosome chr3 = m_outputLayer.getChromosome();

    chromosome.insert(chromosome.end(), chr1.begin(), chr1.end());
    chromosome.insert(chromosome.end(), chr2.begin(), chr2.end());
    chromosome.insert(chromosome.end(), chr3.begin(), chr3.end());

    return chromosome;
}

unsigned DualMLP::getChromosomeSize() const
{
    unsigned chrSize = 0;
    chrSize += m_firstHiddenLayer.getChromosomeSize();
    chrSize += m_secondHiddenLayer.getChromosomeSize();
    chrSize += m_outputLayer.getChromosomeSize();

    return chrSize;
}

std::vector<float> DualMLP::io(const std::vector<float> &inputs)
{
    std::vector<float> response;

    if (inputs.size() != m_inputCount)
    {
        return response;
    }

    response = m_firstHiddenLayer.io(inputs);
    response = m_secondHiddenLayer.io(response);
    response = m_outputLayer.io(response);

    return response;
}