ITK
4.2.0
Insight Segmentation and Registration Toolkit
Main Page
Related Pages
Modules
Namespaces
Classes
Files
Examples
File List
File Members
ITK
Modules
Numerics
NeuralNetworks
include
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
21
#include "
itkTrainingFunctionBase.h
"
22
23
namespace
itk
24
{
25
namespace
Statistics
26
{
33
template
<
class
TSample,
class
TTargetVector,
class
ScalarType>
34
class
IterativeSupervisedTrainingFunction
:
public
TrainingFunctionBase
<TSample, TTargetVector, ScalarType>
35
{
36
public
:
37
38
typedef
IterativeSupervisedTrainingFunction
Self
;
39
typedef
TrainingFunctionBase<TSample, TTargetVector, ScalarType>
Superclass
;
40
typedef
SmartPointer<Self>
Pointer
;
41
typedef
SmartPointer<const Self>
ConstPointer
;
42
44
itkTypeMacro(
IterativeSupervisedTrainingFunction
,
TrainingFunctionBase
);
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);
55
56
itkSetMacro(Threshold, ScalarType);
57
58
protected
:
59
60
IterativeSupervisedTrainingFunction
();
61
virtual
~IterativeSupervisedTrainingFunction
(){};
62
64
virtual
void
PrintSelf
( std::ostream& os,
Indent
indent )
const
;
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
78
Generated on Tue Jul 10 2012 23:32:05 for ITK by
1.8.1