ITK  4.13.0
Insight Segmentation and Registration Toolkit
itkBatchSupervisedTrainingFunction.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 itkBatchSupervisedTrainingFunction_h
19 #define itkBatchSupervisedTrainingFunction_h
20 
22 
23 namespace itk
24 {
25 namespace Statistics
26 {
33 template<typename TSample, typename TTargetVector, typename ScalarType>
34 class ITK_TEMPLATE_EXPORT BatchSupervisedTrainingFunction : public TrainingFunctionBase<TSample, TTargetVector, ScalarType>
35 {
36 public:
37 
43 
46 
48  itkNewMacro(Self);
49 
50  typedef typename Superclass::NetworkType NetworkType;
51  typedef typename Superclass::InternalVectorType InternalVectorType;
52 
54  void SetNumOfIterations(SizeValueType i);
55 
56  virtual void Train(NetworkType* net, TSample* samples, TTargetVector* targets) ITK_OVERRIDE;
57 
58  itkSetMacro(Threshold, ScalarType);
59 
60 protected:
61 
63  virtual ~BatchSupervisedTrainingFunction() ITK_OVERRIDE {}
64 
66  virtual void PrintSelf( std::ostream& os, Indent indent ) const ITK_OVERRIDE;
67 
68  ScalarType m_Threshold;
69  bool m_Stop; //stop condition
70 };
71 
72 } // end namespace Statistics
73 } // end namespace itk
74 
75 #ifndef ITK_MANUAL_INSTANTIATION
76  #include "itkBatchSupervisedTrainingFunction.hxx"
77 #endif
78 
79 #endif
Light weight base class for most itk classes.
unsigned long SizeValueType
Definition: itkIntTypes.h:143
This is the itkBatchSupervisedTrainingFunction class.
TrainingFunctionBase< TSample, TTargetVector, ScalarType > Superclass
This is the itkTrainingFunctionBase class.
Control indentation during Print() invocation.
Definition: itkIndent.h:49