growing_neurons.h

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-only
// Copyright (C) 2026 Artificial Intelligence Techniques, S.L. and Roberto López González
 
#pragma once
 
#include "opennn/core/opennn_types.h"
#include "opennn/model_selection/selection_algorithm.h"
 
namespace opennn
{
 
class Training;
struct NeuronsSelectionResult;
 
class GrowingNeurons : public SelectionAlgorithm
{
 
public:
 
    enum class StoppingCondition {
        MaximumTime,
        ValidationErrorGoal,
        MaximumEpochs,
        MaximumValidationFailures,
        MaximumNeurons
    };
 
    explicit GrowingNeurons(Training* = nullptr);
 
    const Training* get_training() const noexcept { return training; }
 
    void set(Training*);
    void set_training(Training* new_training) noexcept
    {
        training = new_training;
    }
 
    void set_default();
 
    void set_maximum_neurons(const Index new_maximum_neurons) { maximum_neurons = new_maximum_neurons; }
    void set_minimum_neurons(const Index new_minimum_neurons) { minimum_neurons = new_minimum_neurons; }
 
    void set_neurons_increment(const Index new_neurons_increment) { neurons_increment = new_neurons_increment; }
 
    void set_warm_start(bool new_warm_start) { warm_start = new_warm_start; }
 
    NeuronsSelectionResult perform_neurons_selection();
 
    string get_name() const { return "GrowingNeurons"; }
 
    void from_JSON(const JsonDocument&);
 
    void to_JSON(JsonWriter&) const;
 
    void save(const filesystem::path&) const;
    void load(const filesystem::path&);
 
private:
 
    Index minimum_neurons = 0;
 
    Index maximum_neurons = 0;
 
    Index neurons_increment = 0;
 
    bool warm_start = true;
};
 
struct NeuronsSelectionResult
{
   NeuronsSelectionResult(const Index maximum_epochs = 0);
 
   void resize_history(const Index);
 
   void print() const;
 
   VectorI neurons_number_history;
 
   Index optimal_neurons_number = 1;
 
   VectorR optimal_parameters;
 
   VectorR training_error_history;
 
   VectorR validation_error_history;
 
   float optimum_training_error = 10.0f;
 
   float optimum_validation_error = 10.0f;
 
   optional<GrowingNeurons::StoppingCondition> stopping_condition;
 
   string elapsed_time;
};
 
}