ITK
4.3.0
Insight Segmentation and Registration Toolkit
Main Page
Related Pages
Modules
Namespaces
Classes
Files
Examples
File List
File Members
ITK
Modules
Numerics
NeuralNetworks
include
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
21
#include "
itkTrainingFunctionBase.h
"
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
38
typedef
BatchSupervisedTrainingFunction
Self
;
39
typedef
TrainingFunctionBase<TSample, TTargetVector, ScalarType>
40
Superclass
;
41
typedef
SmartPointer<Self>
Pointer
;
42
typedef
SmartPointer<const Self>
ConstPointer
;
43
45
itkTypeMacro(
BatchSupervisedTrainingFunction
,
TrainingFunctionBase
);
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);
57
58
itkSetMacro(Threshold, ScalarType);
59
60
protected
:
61
62
BatchSupervisedTrainingFunction
();
63
virtual
~BatchSupervisedTrainingFunction
(){};
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
Generated on Sun Dec 9 2012 00:46:34 for ITK by
1.8.2