DeepLearningCpp/NeuralNetworkOptions.hpp at master · daelsepara/DeepLearningCpp · 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
#ifndef NEURAL_NETWORK_OPTIONS_HPP
#define NEURAL_NETWORK_OPTIONS_HPP

class NeuralNetworkOptions
{
public:

	double Alpha;
	int Epochs;
	int Inputs;
	int Nodes;
	int Items;
	int Categories;
	double Tolerance;
	int HiddenLayers;
	bool UseL2 = false;

	NeuralNetworkOptions(double alpha, int epochs, int categories, int inputs, int nodes, int items, double tolerance)
	{
		Alpha = alpha;
		Epochs = epochs;
		Inputs = inputs; // Input layer features (i)
		Nodes = nodes; // Hidden layer nodes (j)
		Items = items;	// number of input items
		Categories = categories; // number of output categories (k)
		Tolerance = tolerance;
	}

	NeuralNetworkOptions(double alpha, int epochs, int categories, int inputs, int nodes, int items, double tolerance, bool useL2)
	{
		Alpha = alpha;
		Epochs = epochs;
		Inputs = inputs; // Input layer features (i)
		Nodes = nodes; // Hidden layer nodes (j)
		Items = items;	// number of input items
		Categories = categories; // number of output categories (k)
		Tolerance = tolerance;
		UseL2 = useL2;
	}

	NeuralNetworkOptions(double alpha, int epochs, int categories, int inputs, int items, double tolerance, int hiddenLayers, bool useL2)
	{
		Alpha = alpha;
		Epochs = epochs;
		Inputs = inputs; // Input layer features (i)
		Items = items;	// number of input items
		Categories = categories; // number of output categories (k)
		Tolerance = tolerance;
		HiddenLayers = hiddenLayers; // number of hidden layers
		UseL2 = useL2;
	}

	NeuralNetworkOptions()
	{
		NeuralNetworkOptions(1.0, 1, 2, 2, 16, 50, 0.001);
	}
};

#endif