Inheritance diagram for DefectTrainer:
Public Member Functions | |
DefectTrainer (const DefectixOptions::Options &options) | |
Standard constructor. | |
const errorCodes | exec () |
Main function. | |
const errorCodes | createNetworks () |
create networks recursively | |
Private Member Functions | |
errorCodes | createVectors (gsl_matrix *input, gsl_matrix *output, std::map< int, gsl_matrix * > &inputSrc, std::map< int, gsl_matrix * > &outputSrc, FitsManager &mgr) |
create vectors needed for training | |
void | createKLBase (gsl_matrix *inputV) |
compute or load a KL-Base and compute PCA | |
const errorCodes | createNetwork (gsl_matrix *inputV, gsl_matrix *outputV) |
create a network and train it | |
Private Attributes | |
t_network * | _l0 |
Neural network. | |
int | _nbPC |
Number of principal components. | |
int | _nbVectors |
Number of vectors of the training base. | |
unsigned | _maxIter |
Maximum number of iterations. | |
double | _energyTreshold |
Network energy treshold. | |
std::string | _evecFile |
Name of the file containing the KL-Base. |
The class is the training part of the project.
|
Standard constructor. The constructor calls Defectix constructor and then initialises private members with 'options' values.
|
|
compute or load a KL-Base and compute PCA The functions computes PCA. If a file DefectTrainer::_evecFile exists, it is loaded as the KL-Base. If no file exists, the base is computed first.
|
|
create a network and train it The function creates a network accordingly to input and output vectors. The network is then trained using rprop algorithm.
|
|
create networks recursively The functions creates networks and trains them recursively |
|
create vectors needed for training The functions reads images and masks and create input and output vectors for training.
|
|
Main function. The function starts the training Implements Defectix. |
|
Network energy treshold. _energyTreshold is the energy limit the network must reach to stop converging. |
|
Name of the file containing the KL-Base. _evecFile is the name of the KL-Base stored as a gsl_matrix. |
|
Neural network. _l0 is the neural network used in DefectTrainer::createNetwork() |
|
Maximum number of iterations. _maxIter is the maximum number of iterations the network is allowed to do without any energy decrease. This parameter avoid an infinite loop if _energyTreshold cannot be reached. |
|
Number of principal components. _nbPC is the number of principal components to use once the KL-Base is computed. |
|
Number of vectors of the training base. _nbVectors is the number of vectors create in the input and output sets to train the network |