00001 /*========================================================================= 00002 00003 Program: Insight Segmentation & Registration Toolkit 00004 Module: $RCSfile: itkErrorBackPropagationLearningWithMomentum.h,v $ 00005 Language: C++ 00006 Date: $Date: 2009-01-24 21:33:49 $ 00007 Version: $Revision: 1.4 $ 00008 00009 Copyright (c) Insight Software Consortium. All rights reserved. 00010 See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details. 00011 00012 This software is distributed WITHOUT ANY WARRANTY; without even 00013 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 00014 PURPOSE. See the above copyright notices for more information. 00015 00016 =========================================================================*/ 00017 00026 #ifndef __itkErrorBackPropagationLearningWithMomentum_h 00027 #define __itkErrorBackPropagationLearningWithMomentum_h 00028 00029 #include <iostream> 00030 #include "itkLightProcessObject.h" 00031 #include "itkLearningFunctionBase.h" 00032 00033 namespace itk 00034 { 00035 namespace Statistics 00036 { 00037 00038 template<class LayerType, class TTargetVector> 00039 class ErrorBackPropagationLearningWithMomentum : public LearningFunctionBase<typename LayerType::LayerInterfaceType, TTargetVector> 00040 { 00041 public: 00042 00043 typedef ErrorBackPropagationLearningWithMomentum Self; 00044 typedef LearningFunctionBase<typename LayerType::LayerInterfaceType, TTargetVector> 00045 Superclass; 00046 typedef SmartPointer<Self> Pointer; 00047 typedef SmartPointer<const Self> ConstPointer; 00048 00050 itkTypeMacro(ErrorBackPropagationLearningWithMomentum, LearningFunctionBase); 00051 00053 itkNewMacro(Self); 00054 00055 typedef typename Superclass::ValueType ValueType; 00056 typedef typename LayerType::LayerInterfaceType LayerInterfaceType; 00057 00058 virtual void Learn( LayerInterfaceType * layer, ValueType learningrate ); 00059 virtual void Learn( LayerInterfaceType * layer, TTargetVector errors, ValueType learningrate ); 00060 00061 00062 protected: 00063 ErrorBackPropagationLearningWithMomentum(); 00064 virtual ~ErrorBackPropagationLearningWithMomentum() {}; 00065 00066 virtual void PrintSelf( std::ostream& os, Indent indent ) const; 00067 00068 ValueType m_Momentum; 00069 }; 00070 00071 } // end namespace Statistics 00072 } // end namespace itk 00073 00074 #ifndef ITK_MANUAL_INSTANTIATION 00075 #include "itkErrorBackPropagationLearningWithMomentum.txx" 00076 #endif 00077 00078 #endif 00079