ITK  4.2.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<class TSample, class TTargetVector, class ScalarType>
34 class BatchSupervisedTrainingFunction : public TrainingFunctionBase<TSample, TTargetVector, ScalarType>
35 {
36 public:
37 
43 
46 
48  itkNewMacro(Self);
49 
50  typedef typename Superclass::NetworkType NetworkType;
52 
55 
56  virtual void Train(NetworkType* net, TSample* samples, TTargetVector* targets);
57 
58  itkSetMacro(Threshold, ScalarType);
59 
60 protected:
61 
64 
66  virtual void PrintSelf( std::ostream& os, Indent indent ) const;
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
80