dropout/xor.h at master · turkdogan/dropout · 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
#ifndef XOR_H
#define XOR_H

#include "utils.h"
#include "network.h"

void runXorLayers() {
	std::cout << "Running XOR example..." << std::endl;
	Eigen::MatrixXf input(4, 2);
	input(0, 0) = 0;
	input(0, 1) = 0;
	input(1, 0) = 0;
	input(1, 1) = 1;
	input(2, 0) = 1;
	input(2, 1) = 0;
	input(3, 0) = 1;
	input(3, 1) = 1;

	Eigen::MatrixXf output(4, 1);
	output(0, 0) = 0;
	output(1, 0) = 1;
	output(2, 0) = 1;
	output(3, 0) = 0;

    NetworkConfig config;
    config.epoch_count = 30000;
    config.learning_rate = 0.1f;
    config.momentum = 0.9f;
    config.batch_size = 1;
    config.report_each = 1000;

    config.addLayerConfig(2, 15, Activation::Sigmoid);
    config.addLayerConfig(15, 5, Activation::Sigmoid);
    config.addLayerConfig(5, 1, Activation::Sigmoid);

    Scenario scenario = createNoScenario();

    Network network(scenario, config);
    TrainingResult scenario_result = network.trainNetwork(
        input, output,
        input, output);
    int correct = network.test(input, output);
    std::cout << "Correct: " << correct << std::endl;
}


#endif