training_strategy.h

Go to the documentation of this file.

//   OpenNN: Open Neural Networks Library
//   www.opennn.net
//
//   T R A I N I N G   S T R A T E G Y   C L A S S   H E A D E R
//
//   Artificial Intelligence Techniques SL
//   artelnics@artelnics.com
 
#pragma once
 
#include "loss.h"
#include "optimizer.h"
 
namespace opennn
{
 
class Loss;
class Optimizer;
 
struct TrainingResults;

class TrainingStrategy
{
 
public:

    TrainingStrategy(NeuralNetwork* = nullptr, Dataset* = nullptr);
 
    const Dataset* get_dataset() const { return dataset; }
    Dataset* get_dataset() { return dataset; }
 
    const NeuralNetwork* get_neural_network() const { return neural_network; }
    NeuralNetwork* get_neural_network() { return neural_network; }
 
    const Loss* get_loss() const { return loss.get(); }
    Loss* get_loss() { return loss.get(); }
 
    const Optimizer* get_optimization_algorithm() const { return optimizer.get(); }
    Optimizer* get_optimization_algorithm() { return optimizer.get(); }
    void set(NeuralNetwork* = nullptr, Dataset* = nullptr);
    void set_default();
 
    void set_dataset(Dataset* new_dataset) { dataset = new_dataset; }
    void set_neural_network(NeuralNetwork* new_neural_network) { neural_network = new_neural_network; }

    void set_loss(const string&);
    void set_optimization_algorithm(const string&);

    TrainingResults train();
    void from_JSON(const JsonDocument&);
    void to_JSON(JsonWriter&) const;

    void save(const filesystem::path&) const;
    void load(const filesystem::path&);
 
private:
 
    void fix_forecasting();
 
    Dataset* dataset = nullptr;
 
    NeuralNetwork* neural_network = nullptr;
 
    unique_ptr<Loss> loss;
 
    unique_ptr<Optimizer> optimizer;
};
 
}
 
// OpenNN: Open Neural Networks Library.
// Copyright(C) 2005-2026 Artificial Intelligence Techniques, SL.
// Licensed under the GNU Lesser General Public License v2.1 or later.