openGPMP
Open Source Mathematics Package
|
MDLAutoEncoder Minimal Description Length class, a derived class from AutoEncoder. More...
#include <encoder.hpp>
Public Member Functions | |
MDLAutoEncoder (int input_size, int hidden_size, int output_size, double learning_rate, double mdl_weight) | |
Constructor for the MDLAutoEncoder class. More... | |
virtual void | train (const std::vector< std::vector< double >> &training_data, int epochs) override |
Trains the MDL autoencoder on the given training data. 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 | mdl_weight |
Weight for the MDL penalty term. 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... | |
MDLAutoEncoder Minimal Description Length class, a derived class from AutoEncoder.
Definition at line 314 of file encoder.hpp.
gpmp::ml::MDLAutoEncoder::MDLAutoEncoder | ( | int | input_size, |
int | hidden_size, | ||
int | output_size, | ||
double | learning_rate, | ||
double | mdl_weight | ||
) |
Constructor for the MDLAutoEncoder 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 |
mdl_weight | The weight for the MDL penalty term |
Definition at line 378 of file encoder.cpp.
|
overridevirtual |
Trains the MDL autoencoder on the given training data.
Overrides the train method in the base class with MDL penalty
training_data | The training data |
epochs | The number of training epochs |
Reimplemented from gpmp::ml::AutoEncoder.
Definition at line 386 of file encoder.cpp.
double gpmp::ml::MDLAutoEncoder::mdl_weight |
Weight for the MDL penalty term.
Definition at line 319 of file encoder.hpp.