GeneticProgramming/Mutation.h at main · MichalicekPetr/GeneticProgramming · GitHub

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
#pragma once

#include <string>

#include "Individual.h"
#include "Node.h"

using namespace std;

class Mutation
{
	string name;

public:
	virtual void mutate(Individual & individual, const int& maxDepth) = 0;
	virtual void setTerminalSet(TerminalSet termSet) = 0;
};


class SubtreeMutation : public Mutation
{
private:
	string name = "Sub tree mutation";
	double mutationProb;
	FunctionSet funcSet;
	TerminalSet termSet;

public:
	SubtreeMutation();
	SubtreeMutation(const double& mutationProb, const FunctionSet& funcSet, const TerminalSet& termSet);
	void mutate(Individual & individual, const int& maxDepth) override;
	void setTerminalSet(TerminalSet termSet) override;
};


class NodeReplacementMutation : public Mutation
{
private:
	string name = "Node replacement mutation";
	double mutationProb;
	FunctionSet funcSet;
	TerminalSet termSet;

public:
	NodeReplacementMutation();
	NodeReplacementMutation(const double & mutationProb, const FunctionSet & funcSet, const TerminalSet & termSet);
	void setTerminalSet(TerminalSet termSet) override;
	void mutate(Individual & individual, const int & maxDepth = 0) override;
};


class CombinedMutation : public Mutation {
	string name = "Combined mutation";
	double mutationProb;
	FunctionSet funcSet;
	TerminalSet termSet;
	SubtreeMutation subTreeMutation;
	NodeReplacementMutation nodeReplacementMutation;

public:
	CombinedMutation(const double& nodeReplacementMutationProb, const double& subTreeMutationProb, const FunctionSet& funcSet, const TerminalSet& termSet);
	void mutate(Individual& individual, const int & maxDepth) override;
	void setTerminalSet(TerminalSet termSet) override;
};