DBN__destruct

Yusuke Sugomori · Yusuke Sugomori · commit c4dbea6b198a · 2013-02-12T00:32:34.000+09:00
diff --git a/c/DBN.c b/c/DBN.c
@@ -74,7 +74,13 @@ void DBN__construct(DBN* this, int N, \
 }
 
 void DBN__destruct(DBN* this) {
-  
+  int i;
+  for(i=0; i<this->n_layers; i++) {
+    HiddenLayer__destruct(&(this->sigmoid_layers[i]));
+    RBM__destruct(&(this->rbm_layers[i]));
+  }
+  free(this->sigmoid_layers);
+  free(this->rbm_layers);
 }
 
 void DBN_pretrain(DBN* this, int *input, double lr, int k, int epochs) {
@@ -561,13 +567,15 @@ void test_dbn(void) {
     }
     printf("\n");
   }
+
+  // destruct DBN
+  DBN__destruct(&dbn);
   
 }
 
 
 
 int main(void) {
   test_dbn();
-  
   return 0;
 }

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,13 @@ void DBN__construct(DBN* this, int N, \`
`74`	`74`	`}`
`75`	`75`
`76`	`76`	`void DBN__destruct(DBN* this) {`
`77`		`-`
	`77`	`+ int i;`
	`78`	`+ for(i=0; i<this->n_layers; i++) {`
	`79`	`+ HiddenLayer__destruct(&(this->sigmoid_layers[i]));`
	`80`	`+ RBM__destruct(&(this->rbm_layers[i]));`
	`81`	`+ }`
	`82`	`+ free(this->sigmoid_layers);`
	`83`	`+ free(this->rbm_layers);`
`78`	`84`	`}`
`79`	`85`
`80`	`86`	`void DBN_pretrain(DBN* this, int *input, double lr, int k, int epochs) {`
`@@ -561,13 +567,15 @@ void test_dbn(void) {`
`561`	`567`	`}`
`562`	`568`	`printf("\n");`
`563`	`569`	`}`
	`570`	`+`
	`571`	`+ // destruct DBN`
	`572`	`+ DBN__destruct(&dbn);`
`564`	`573`
`565`	`574`	`}`
`566`	`575`
`567`	`576`
`568`	`577`
`569`	`578`	`int main(void) {`
`570`	`579`	`test_dbn();`
`571`		`-`
`572`	`580`	`return 0;`
`573`	`581`	`}`