model_selection.h

//   OpenNN: Open Neural Networks Library
//   www.opennn.net
//
//   M O D E L   S E L E C T I O N   C L A S S   H E A D E R               
//
//   Artificial Intelligence Techniques SL
//   artelnics@artelnics.com
 
#ifndef MODELSELECTION_H
#define MODELSELECTION_H
 
// System includes
 
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <cmath>
#include <ctime>
 
// OpenNN includes
 
#include "config.h"
#include "training_strategy.h"
#include "growing_neurons.h"
#include "growing_inputs.h"
#include "pruning_inputs.h"
#include "genetic_algorithm.h"
 
namespace OpenNN
{
 
 
 
class ModelSelection
{
 
public:  
 
    // Constructors
 
    explicit ModelSelection();
 
    explicit ModelSelection(TrainingStrategy*);
 
    // Destructor
 
    virtual ~ModelSelection();
 
 
    enum class NeuronsSelectionMethod{GROWING_NEURONS};
 
 
    enum class InputsSelectionMethod{GROWING_INPUTS, PRUNING_INPUTS, GENETIC_ALGORITHM};
 
    // Get methods
 
    TrainingStrategy* get_training_strategy_pointer() const;
    bool has_training_strategy() const;
 
    const NeuronsSelectionMethod& get_neurons_selection_method() const;
    const InputsSelectionMethod& get_inputs_selection_method() const;
 
    GrowingNeurons* get_growing_neurons_pointer();
 
    GrowingInputs* get_growing_inputs_pointer();
    PruningInputs* get_pruning_inputs_pointer();
    GeneticAlgorithm* get_genetic_algorithm_pointer();
 
    // Set methods
 
    void set(TrainingStrategy*);
 
    void set_default();
 
    void set_display(const bool&);
 
    void set_neurons_selection_method(const NeuronsSelectionMethod&);
    void set_neurons_selection_method(const string&);
 
    void set_inputs_selection_method(const InputsSelectionMethod&);
    void set_inputs_selection_method(const string&);
 
    // Model selection methods
 
    void check() const;
 
    NeuronsSelectionResults perform_neurons_selection();
 
    InputsSelectionResults perform_inputs_selection();
 
    // Serialization methods
    
    void from_XML(const tinyxml2::XMLDocument&);
 
    void write_XML(tinyxml2::XMLPrinter&) const;
 
    string write_neurons_selection_method() const;
    string write_inputs_selection_method() const;
 
    void print() const;
    void save(const string&) const;
    void load(const string&);
 
private: 
 
 
    TrainingStrategy* training_strategy_pointer = nullptr;
 
 
    GrowingNeurons growing_neurons;
 
 
    GrowingInputs growing_inputs;
 
 
    PruningInputs pruning_inputs;
 
 
    GeneticAlgorithm genetic_algorithm;
 
 
    NeuronsSelectionMethod neurons_selection_method;
 
 
    InputsSelectionMethod inputs_selection_method;
 
 
    bool display = true;
};
 
}
 
#endif