diff options
Diffstat (limited to '')
| -rw-r--r-- | src/row.h | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/src/row.h b/src/row.h new file mode 100644 index 0000000..0ec5dde --- /dev/null +++ b/src/row.h | |||
| @@ -0,0 +1,81 @@ | |||
| 1 | #ifndef NEURAL_ROW_H | ||
| 2 | #define NEURAL_ROW_H | ||
| 3 | |||
| 4 | #include "neural/container.h" | ||
| 5 | |||
| 6 | namespace Neural | ||
| 7 | { | ||
| 8 | template<typename sigtype> | ||
| 9 | class Row : public Container<sigtype> | ||
| 10 | { | ||
| 11 | public: | ||
| 12 | //typedef Bu::List<Neural::Node<sigtype> *> NodeList; | ||
| 13 | Row() : | ||
| 14 | iInputs( 0 ), | ||
| 15 | iOutputs( 0 ), | ||
| 16 | iWeights( 0 ), | ||
| 17 | iBiases( 0 ) | ||
| 18 | { | ||
| 19 | } | ||
| 20 | |||
| 21 | virtual ~Row() | ||
| 22 | { | ||
| 23 | } | ||
| 24 | |||
| 25 | virtual void finalize( int iNumInputs ) | ||
| 26 | { | ||
| 27 | iInputs = iNumInputs; | ||
| 28 | iOutputs = 0; | ||
| 29 | iWeights = 0; | ||
| 30 | iBiases = 0; | ||
| 31 | |||
| 32 | for( typename Container<sigtype>::NodeList::iterator i = | ||
| 33 | Container<sigtype>::getNodeList().begin(); i; i++ ) | ||
| 34 | { | ||
| 35 | (*i)->finalize( iInputs ); | ||
| 36 | iOutputs += (*i)->getNumOutputs(); | ||
| 37 | iWeights += (*i)->getNumWeights(); | ||
| 38 | iBiases += (*i)->getNumBiases(); | ||
| 39 | } | ||
| 40 | } | ||
| 41 | |||
| 42 | virtual void process( sigtype *aInput, sigtype *aOutput ) | ||
| 43 | { | ||
| 44 | int iOutputOffset = 0; | ||
| 45 | for( typename Container<sigtype>::NodeList::iterator i = | ||
| 46 | Container<sigtype>::getNodeList().begin(); i; i++ ) | ||
| 47 | { | ||
| 48 | (*i)->process( aInput, aOutput+iOutputOffset ); | ||
| 49 | iOutputOffset += (*i)->getNumOutputs(); | ||
| 50 | } | ||
| 51 | } | ||
| 52 | |||
| 53 | virtual int getNumInputs() const | ||
| 54 | { | ||
| 55 | return iInputs; | ||
| 56 | } | ||
| 57 | |||
| 58 | virtual int getNumOutputs() const | ||
| 59 | { | ||
| 60 | return iOutputs; | ||
| 61 | } | ||
| 62 | |||
| 63 | virtual int getNumWeights() const | ||
| 64 | { | ||
| 65 | return iWeights; | ||
| 66 | } | ||
| 67 | |||
| 68 | virtual int getNumBiases() const | ||
| 69 | { | ||
| 70 | return iBiases; | ||
| 71 | } | ||
| 72 | |||
| 73 | private: | ||
| 74 | int iInputs; | ||
| 75 | int iOutputs; | ||
| 76 | int iWeights; | ||
| 77 | int iBiases; | ||
| 78 | }; | ||
| 79 | }; | ||
| 80 | |||
| 81 | #endif | ||
