Newer
Older
Cédric Traizet
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#ifndef cbTrainAutoencoder_txx
#define cbTrainAutoencoder_txx
#include "cbLearningApplicationBaseDR.h"
namespace otb
{
namespace Wrapper
{
template <class TInputValue, class TOutputValue>
void
cbLearningApplicationBaseDR<TInputValue,TOutputValue>
::InitAutoencoderParams()
{
AddChoice("model.tiedautoencoder", "Shark Tied Autoencoder");
AddChoice("model.autoencoder", "Shark Autoencoder");
SetParameterDescription("model.autoencoder",
"This group of parameters allows setting Shark autoencoder parameters. "
);
//Number Of Iterations
AddParameter(ParameterType_Int, "model.autoencoder.nbiter",
"Maximum number of iterations during training");
SetParameterInt("model.autoencoder.nbiter",100, false);
SetParameterDescription(
"model.autoencoder.nbiter",
"The maximum number of iterations used during training.");
//Number Of Hidden Neurons
AddParameter(ParameterType_Int, "model.autoencoder.nbneuron",
"Number of neurons in the hidden layer");
SetParameterInt("model.autoencoder.nbneuron",10, false);
SetParameterDescription(
"model.autoencoder.nbneuron",
"The number of neurons in the hidden layer.");
//normalization
AddParameter(ParameterType_Float, "model.autoencoder.normalizer", "Strength of the normalization");
SetParameterFloat("model.autoencoder.normalizer",0, false);
SetParameterDescription("model.autoencoder.normalizer",
"Strength of the L2 normalization used during training");
}
Cédric Traizet
committed
template <class TInputValue, class TOutputValue>
Cédric Traizet
committed
void cbLearningApplicationBaseDR<TInputValue,TOutputValue>
::TrainAutoencoder(typename ListSampleType::Pointer trainingListSample,std::string modelPath)
Cédric Traizet
committed
{
typename autoencoderchoice::Pointer dimredTrainer = autoencoderchoice::New();
Cédric Traizet
committed
dimredTrainer->SetNumberOfHiddenNeurons(GetParameterInt("model.autoencoder.nbneuron"));
dimredTrainer->SetNumberOfIterations(GetParameterInt("model.autoencoder.nbiter"));
dimredTrainer->SetRegularization(GetParameterFloat("model.autoencoder.normalizer"));
dimredTrainer->SetInputListSample(trainingListSample);
dimredTrainer->Train();
dimredTrainer->Save(modelPath);
}
Cédric Traizet
committed
} //end namespace wrapper
} //end namespace otb
#endif