Model.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/TrainingSpec.hpp"
#include "utils/Types.hpp"
 
#include <memory>
 
namespace ppforest2 {
  struct Tree;
  struct Forest;

  struct Proportions {};
 

  struct Model {
    using Ptr = std::shared_ptr<Model>;

    struct Visitor {
      virtual void visit(Tree const& tree)     = 0;
      virtual void visit(Forest const& forest) = 0;
    };
 
    virtual ~Model() = default;

    bool degenerate = false;

    TrainingSpec::Ptr training_spec;

    virtual void accept(Visitor& visitor) const = 0;

    virtual types::Response predict(types::FeatureVector const& data) const = 0;

    virtual types::ResponseVector predict(types::FeatureMatrix const& data) const = 0;

    virtual types::FeatureMatrix predict(types::FeatureMatrix const& data, Proportions) const = 0;

    static Ptr train(TrainingSpec const& spec, types::FeatureMatrix const& x, types::ResponseVector const& y);
  };
}