selection_algorithm.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"
 
namespace opennn
{
 
class Training;
 
class SelectionAlgorithm
{
public:
 
    explicit SelectionAlgorithm(Training* new_training = nullptr)
        : training(new_training)
    {
    }
 
    virtual ~SelectionAlgorithm() = default;
 
    const Training* get_training() const noexcept { return training; }
 
    void set(Training* new_training) { training = new_training; }
 
    void set_trials_number(const Index new_trials_number) { trials_number = new_trials_number; }
 
    void set_display(bool new_display) { display = new_display; }
 
    void set_validation_error_goal(const float new_validation_error_goal) { validation_error_goal = new_validation_error_goal; }
    void set_maximum_epochs(const Index new_maximum_epochs) { maximum_epochs = new_maximum_epochs; }
    void set_maximum_validation_failures(const Index new_maximum_validation_failures) { maximum_validation_failures = new_maximum_validation_failures; }
    void set_maximum_time(const float new_maximum_time) { maximum_time = new_maximum_time; }
 
    void set_folds_number(const Index new_folds_number) { folds_number = max<Index>(new_folds_number, Index(1)); }
 
protected:
 
    Training* training = nullptr;
 
    Index trials_number = 1;
 
    Index folds_number = 1;
 
    bool display = true;
 
    float validation_error_goal = 0;
 
    Index maximum_epochs = 10;
 
    Index maximum_validation_failures = 100;
 
    float maximum_time = 0;
};
 
}