gp/src/problem.h

Go to the documentation of this file.

 
#if !defined(VITA_SRC_PROBLEM_H)
#define      VITA_SRC_PROBLEM_H
 
#include <filesystem>
#include <string>
 
#include "kernel/exceptions.h"
#include "kernel/problem.h"
#include "kernel/gp/src/category_set.h"
#include "kernel/gp/src/primitive/factory.h"
 
namespace vita
{
 
enum class dataset_t {training = 0, validation, test};
 
class src_problem : public problem
{
public:
  // --- Constructors ---
  src_problem();
 
  explicit src_problem(const std::filesystem::path &, typing = typing::weak);
  explicit src_problem(std::istream &, typing = typing::weak);
 
  struct default_symbols_t {};
  static const default_symbols_t default_symbols;
  src_problem(const std::filesystem::path &, const default_symbols_t &,
              typing = typing::weak);
  src_problem(const std::filesystem::path &, const std::filesystem::path &,
              typing = typing::weak);
  // --------------------
 
  bool operator!() const;
  std::size_t setup_symbols(typing = typing::weak);
  std::size_t setup_symbols(const std::filesystem::path &,
                            typing = typing::weak);
  void setup_terminals(typing);
 
  const dataframe &data(dataset_t = dataset_t::training) const;
  dataframe &data(dataset_t = dataset_t::training);
 
  bool classification() const { return classes() > 1; }
 
  unsigned categories() const;
  unsigned classes() const;
  unsigned variables() const;
 
  bool is_valid() const override;
 
private:
  // Private support methods.
  bool compatible(const cvect &, const std::vector<std::string> &,
                  const category_set &) const;
  std::size_t setup_symbols_impl();
  std::size_t setup_symbols_impl(const std::filesystem::path &);
 
  // Private data members.
  dataframe training_;
  dataframe validation_;
  symbol_factory factory_;
};
 
}  // namespace vita
 
#endif  // include guard