Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/moses-smt/nplm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/graphClasses.h')
-rw-r--r--src/graphClasses.h89
1 files changed, 44 insertions, 45 deletions
diff --git a/src/graphClasses.h b/src/graphClasses.h
index d3c0c4a..cd80a4c 100644
--- a/src/graphClasses.h
+++ b/src/graphClasses.h
@@ -3,7 +3,6 @@
#include <cstdlib>
#include "neuralClasses.h"
-//#include <../3rdparty/Eigen/Dense>
#include <Eigen/Dense>
namespace nplm
@@ -11,50 +10,50 @@ namespace nplm
template <class X>
class Node {
- public:
- X * param; //what parameter is this
- //vector <void *> children;
- //vector <void *> parents;
- Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> fProp_matrix;
- Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> bProp_matrix;
- int minibatch_size;
-
- public:
- Node() : param(NULL), minibatch_size(0) { }
-
- Node(X *input_param, int minibatch_size)
- : param(input_param),
- minibatch_size(minibatch_size)
- {
- resize(minibatch_size);
- }
-
- void resize(int minibatch_size)
- {
- this->minibatch_size = minibatch_size;
- if (param->n_outputs() != -1)
- {
- fProp_matrix.setZero(param->n_outputs(), minibatch_size);
- }
- if (param->n_inputs() != -1)
- {
- bProp_matrix.setZero(param->n_inputs(), minibatch_size);
- }
- }
-
- void resize() { resize(minibatch_size); }
-
- /*
- void Fprop(Matrix<double,Dynamic,Dynamic> & input,int n_cols)
- {
- param->fProp(input,fProp_matrix,0,0,n_cols);
- }
- void Fprop(Matrix<double,1,Dynamic> & input,int n_cols)
- {
- param->fProp(input,fProp_matrix,0,0,n_cols);
- }
- */
- //for f prop, just call the fProp node of the particular parameter.
+ public:
+ X * param; //what parameter is this
+ //vector <void *> children;
+ //vector <void *> parents;
+ Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> fProp_matrix;
+ Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> bProp_matrix;
+ int minibatch_size;
+
+ public:
+ Node() : param(NULL), minibatch_size(0) { }
+
+ Node(X *input_param, int minibatch_size)
+ : param(input_param),
+ minibatch_size(minibatch_size)
+ {
+ resize(minibatch_size);
+ }
+
+ void resize(int minibatch_size)
+ {
+ this->minibatch_size = minibatch_size;
+ if (param->n_outputs() != -1)
+ {
+ fProp_matrix.setZero(param->n_outputs(), minibatch_size);
+ }
+ if (param->n_inputs() != -1)
+ {
+ bProp_matrix.setZero(param->n_inputs(), minibatch_size);
+ }
+ }
+
+ void resize() { resize(minibatch_size); }
+
+ /*
+ void Fprop(Matrix<double,Dynamic,Dynamic> & input,int n_cols)
+ {
+ param->fProp(input,fProp_matrix,0,0,n_cols);
+ }
+ void Fprop(Matrix<double,1,Dynamic> & input,int n_cols)
+ {
+ param->fProp(input,fProp_matrix,0,0,n_cols);
+ }
+ */
+ //for f prop, just call the fProp node of the particular parameter.
};