-
Notifications
You must be signed in to change notification settings - Fork 0
/
NeuralNetworks.h
59 lines (54 loc) · 1001 Bytes
/
NeuralNetworks.h
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
#ifndef NN
#define NN
#include "Matrix.h"
#include <vector>
using std::vector;
#include <iostream>
#include "NNLib.h"
class NeuralNetworks
{
public:
NeuralNetworks();
~NeuralNetworks();
void loadData(const char *srcFile);
void loadX(const char *srcFile);
void loadY(const char *srcFile);
void loadTheta(const char *srcFile, Matrix &thetaI);
bool loadedData();
bool inited_Layer();
void setNumLayers(int num){
numLayer = num;
theta.createNode(num);
grad.createNode(num);
}
Matrix& getThetaI(int i){
return theta[i];
}
Matrix& getY(){
return y;
}
void setNumOutputs(int numOuts){
numOutputs = numOuts;
}
double trainAcurrateNN();
Matrix randInitTheta(int Lin, int Lout);
void computeJreg(double lambda);
double getJreg(){
return Jreg;
}
void updateGrad(double lambda);
Matrix& getGradI(int i){
return grad[i];
}
private:
VectorMatrix theta;
VectorMatrix grad;
Matrix X;
Matrix y;
int mX;
int nX;
int numLayer;
int numOutputs;
double Jreg;
};
#endif