ITK  4.13.0
Insight Segmentation and Registration Toolkit
itkIterativeSupervisedTrainingFunction.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright Insight Software Consortium
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 #ifndef itkIterativeSupervisedTrainingFunction_h
19 #define itkIterativeSupervisedTrainingFunction_h
20 
22 
23 namespace itk
24 {
25 namespace Statistics
26 {
33 template<typename TSample, typename TTargetVector, typename ScalarType>
34 class ITK_TEMPLATE_EXPORT IterativeSupervisedTrainingFunction : public TrainingFunctionBase<TSample, TTargetVector, ScalarType>
35 {
36 public:
37 
42 
45 
47  itkNewMacro(Self);
48 
49  typedef typename Superclass::NetworkType NetworkType;
50  typedef typename Superclass::InternalVectorType InternalVectorType;
51 
52  void SetNumOfIterations(SizeValueType i);
53 
54  virtual void Train(NetworkType* net, TSample* samples, TTargetVector* targets) ITK_OVERRIDE;
55 
56  itkSetMacro(Threshold, ScalarType);
57 
58 protected:
59 
61  virtual ~IterativeSupervisedTrainingFunction() ITK_OVERRIDE{};
62 
64  virtual void PrintSelf( std::ostream& os, Indent indent ) const ITK_OVERRIDE;
65 
66  ScalarType m_Threshold;
67  bool m_Stop; //stop condition
68 };
69 
70 } // end namespace Statistics
71 } // end namespace itk
72 
73 #ifndef ITK_MANUAL_INSTANTIATION
74  #include "itkIterativeSupervisedTrainingFunction.hxx"
75 #endif
76 
77 #endif
Light weight base class for most itk classes.
unsigned long SizeValueType
Definition: itkIntTypes.h:143
TrainingFunctionBase< TSample, TTargetVector, ScalarType > Superclass
This is the itkIterativeSupervisedTrainingFunction class.
This is the itkTrainingFunctionBase class.
Control indentation during Print() invocation.
Definition: itkIndent.h:49