MNISTPatchSoftmaxApplication.hpp

Go to the documentation of this file.

#ifndef MNISTPATCHSOFTMAXAPPLICATION_H_
#define MNISTPATCHSOFTMAXAPPLICATION_H_


#include <opengl/application/OpenGLContinuousLearningApplication.hpp>

#include <opengl/visualization/WindowMatrix2D.hpp>
#include <opengl/visualization/WindowProbability.hpp>
#include <opengl/visualization/WindowCollectorChart.hpp>
using namespace mic::opengl::visualization;

#include <importers/MNISTPatchImporter.hpp>

#include <mlnn/BackpropagationNeuralNetwork.hpp>
using namespace mic::mlnn;

#include <encoders/MatrixXfMatrixXfEncoder.hpp>
#include <encoders/UIntMatrixXfEncoder.hpp>


namespace mic {
namespace applications {

class MNISTPatchSoftmaxApplication: public mic::opengl::application::OpenGLContinuousLearningApplication {
public:
    MNISTPatchSoftmaxApplication(std::string node_name_ = "mnist_patch_autoencoder_reconstruction");

    virtual ~MNISTPatchSoftmaxApplication();

protected:
    virtual void initializePropertyDependentVariables();

    virtual void initialize(int argc, char* argv[]);

    virtual bool performLearningStep();

    virtual void collectTestStatistics();

    virtual void populateTestStatistics();


private:
    mic::importers::MNISTPatchImporter* training_dataset_importer;

    mic::importers::MNISTPatchImporter* test_dataset_importer;

    mic::encoders::MatrixXfMatrixXfEncoder* mnist_encoder;

    mic::encoders::UIntMatrixXfEncoder* label_encoder;

    WindowMatrix2D* w2d_input;

    WindowCollectorChart<float>* w_chart;

    WindowProbability* w_prob;

    mic::types::MatrixXfPtr input_image;

    mic::types::MatrixXfPtr decoded_prediction;

    mic::types::MatrixXfPtr input_target;

    size_t patch_size;

    mic::utils::DataCollectorPtr<std::string, float> collector_ptr;

    mic::configuration::Property<std::string> autoencoder_filename;

    mic::configuration::Property<size_t> autoencoder_layers_to_be_removed;

    mic::configuration::Property<std::string> softmax_filename;

    mic::configuration::Property<bool> softmax_save;

    mic::configuration::Property<bool> softmax_load;

    BackpropagationNeuralNetwork<float> neural_net;

};

} /* namespace applications */
} /* namespace mic */

#endif /* MNISTPATCHSOFTMAXAPPLICATION_H_ */