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 |
1.3.5