StackedAutoEncoder/StackedAutoEncoder.cpp at master · mmx5/StackedAutoEncoder · 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

#include "iostream"
#include "Layer.h"
#include "SimpleAutoEncoder.h"
#include "StackedAutoEncoder.h"

using namespace std;

StackedAutoEncoder::StackedAutoEncoder(int number_AE, int inputDim, int* eachEncoder_outputDim) {
	pAutoEncoder = NULL;
	num_AE = 0;

	Init(number_AE, inputDim, eachEncoder_outputDim);
}

void StackedAutoEncoder::Init(int number_AE, int inputDim, int* eachEncoder_outputDim) {
	if (Is_Inited())
		Delete();

	num_AE = number_AE;

	try {
		pAutoEncoder = new SimpleAutoEncoder[num_AE];
	}
	catch (bad_alloc& ba) {
		printf("bad memory allocation in func : %s, file : %s, line number : %d\n", __FUNCTION__, __FILE__, __LINE__);
		exit(-1);
	}

	pAutoEncoder[0].Init(inputDim, eachEncoder_outputDim[0]);
	for (int i = 1; i < num_AE; i++)
		pAutoEncoder[i].Init(eachEncoder_outputDim[i - 1], eachEncoder_outputDim[i]);

}

void StackedAutoEncoder::Delete() {
	if (pAutoEncoder != NULL) {
		delete[] pAutoEncoder;    //이렇게 하면 pAutoEncoder[0] [1] [2] 각각의 pAutoEncoder[0].delete가 소멸자를 통해 호출되는지 확인하기
		pAutoEncoder = NULL;
	}
	num_AE = 0;
}


//train 하는것
//나중에 decode대신 encode로 되는지... 한번 테스트해보기//명심할것 error계산은 decoder결과값으로 함
void StackedAutoEncoder::Back_Propagate(float* pInput, int action_idx) {
	if (action_idx == 0) {
		pAutoEncoder[0].Back_Propagate(pInput);
	}
	else {
		pAutoEncoder[0].Decoding(pInput);;
		for (int i = 1; i < action_idx; i++)
			pAutoEncoder[i].Decoding(pAutoEncoder[i - 1].Get_Encoding_Result());

		pAutoEncoder[action_idx].Back_Propagate(pAutoEncoder[action_idx - 1].Get_Encoding_Result());
	}
}


//나중에 decode대신 encode로 되는지... 한번 테스트해보기//명심할것 error계산은 decoder결과값으로 함
void StackedAutoEncoder::Decoding(float* pInput, int action_idx) {
	if (action_idx == 0) {
		pAutoEncoder[0].Decoding(pInput);
	}
	else {
		pAutoEncoder[0].Decoding(pInput);
		for (int i = 1; i < action_idx; i++)
			pAutoEncoder[i].Decoding(pAutoEncoder[i - 1].Get_Encoding_Result());

		pAutoEncoder[action_idx].Decoding(pAutoEncoder[action_idx - 1].Get_Encoding_Result());
	}
}
//나중에 decode대신 encode로 되는지... 한번 테스트해보기 //명심할것 error계산은 decoder결과값으로 함
void StackedAutoEncoder::Encoding(float* pInput, int action_idx) {
	if (action_idx == 0) {
		pAutoEncoder[0].Encoding(pInput);
	}
	else {
		pAutoEncoder[0].Decoding(pInput);
		for (int i = 1; i < action_idx; i++)
			pAutoEncoder[i].Decoding(pAutoEncoder[i - 1].Get_Encoding_Result());

		pAutoEncoder[action_idx].Encoding(pAutoEncoder[action_idx - 1].Get_Encoding_Result());
	}
}