|
openGPMP
Open Source Mathematics Package
|
DenoisingAutoEncoder class, a derived class from AutoEncoder. More...
#include <encoder.hpp>
Public Member Functions | |
| DenoisingAutoEncoder (int input_size, int hidden_size, int output_size, double learning_rate, double corruption_level) | |
| Constructor for the DenoisingAutoEncoder class. More... | |
| void | train (const std::vector< std::vector< double >> &training_data, int epochs) override |
| Adds noise to the input data and trains the denoising autoencoder. More... | |
Public Member Functions inherited from gpmp::ml::AutoEncoder | |
| AutoEncoder (int input_size, int hidden_size, int output_size, double learning_rate) | |
| Constructor for the AutoEncoder class. More... | |
| std::vector< double > | sigmoid (const std::vector< double > &x) |
| Sigmoid activation function. More... | |
| std::vector< double > | forward (const std::vector< double > &input) |
| Forward pass through the autoencoder. More... | |
| void | lrate_set (double initial_rate) |
| Set the initial learning rate. More... | |
| virtual void | lrate_update (int epoch) |
| Update the learning rate based on a schedule. More... | |
| void | display () |
| Print the weights of the autoencoder. More... | |
| virtual void | save (const std::string &filename) const |
| Save the model weights to a file. More... | |
| virtual void | load (const std::string &filename) |
| Load model weights from a file. More... | |
Public Attributes | |
| double | corruption_level |
| Probability of setting an input value to zero during training (dropout rate) More... | |
Public Attributes inherited from gpmp::ml::AutoEncoder | |
| int | input_size |
| Size of the input layer. More... | |
| int | hidden_size |
| Size of the hidden layer. More... | |
| int | output_size |
| Size of the output layer. More... | |
| double | learning_rate |
| Learning rate for training the autoencoder. More... | |
| std::vector< std::vector< double > > | weights_input_hidden |
| Weights from the input layer to the hidden layer. More... | |
| std::vector< std::vector< double > > | weights_hidden_output |
| Weights from the hidden layer to the output layer. More... | |
DenoisingAutoEncoder class, a derived class from AutoEncoder.
Definition at line 237 of file encoder.hpp.
| gpmp::ml::DenoisingAutoEncoder::DenoisingAutoEncoder | ( | int | input_size, |
| int | hidden_size, | ||
| int | output_size, | ||
| double | learning_rate, | ||
| double | corruption_level | ||
| ) |
Constructor for the DenoisingAutoEncoder class.
| input_size | The size of the input layer |
| hidden_size | The size of the hidden layer |
| output_size | The size of the output layer |
| learning_rate | The learning rate for training |
| corruption_level | The probability of setting an input value to zero during training |
Definition at line 283 of file encoder.cpp.
|
overridevirtual |
Adds noise to the input data and trains the denoising autoencoder.
Overrides the train method in the base class for denoising
| training_data | The training data |
| epochs | The number of training epochs |
Reimplemented from gpmp::ml::AutoEncoder.
Definition at line 292 of file encoder.cpp.
Referenced by main().
| double gpmp::ml::DenoisingAutoEncoder::corruption_level |
Probability of setting an input value to zero during training (dropout rate)
Definition at line 243 of file encoder.hpp.