OTB  6.7.0
Orfeo Toolbox
otbTrainNormalBayes.hxx
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2005-2019 Centre National d'Etudes Spatiales (CNES)
3  *
4  * This file is part of Orfeo Toolbox
5  *
6  * https://www.orfeo-toolbox.org/
7  *
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  * http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  */
20 
21 #ifndef otbTrainNormalBayes_hxx
22 #define otbTrainNormalBayes_hxx
25 
26 namespace otb
27 {
28 namespace Wrapper
29 {
30 
31  template <class TInputValue, class TOutputValue>
32  void
33  LearningApplicationBase<TInputValue,TOutputValue>
34  ::InitNormalBayesParams()
35  {
36  AddChoice("classifier.bayes", "Normal Bayes classifier");
37  SetParameterDescription("classifier.bayes", "http://docs.opencv.org/modules/ml/doc/normal_bayes_classifier.html");
38 
39  }
40 
41  template <class TInputValue, class TOutputValue>
42  void
43  LearningApplicationBase<TInputValue,TOutputValue>
44  ::TrainNormalBayes(typename ListSampleType::Pointer trainingListSample,
45  typename TargetListSampleType::Pointer trainingLabeledListSample,
46  std::string modelPath)
47  {
49  typename NormalBayesType::Pointer classifier = NormalBayesType::New();
50  classifier->SetRegressionMode(this->m_RegressionFlag);
51  classifier->SetInputListSample(trainingListSample);
52  classifier->SetTargetListSample(trainingLabeledListSample);
53  classifier->Train();
54  classifier->Save(modelPath);
55  }
56 
57 } //end namespace wrapper
58 } //end namespace otb
59 
60 #endif