Simulation.hpp

Go to the documentation of this file.

#pragma once
 
#include "stats/Stats.hpp"
#include "stats/DataPacket.hpp"
#include "utils/Types.hpp"
 
#include <vector>
 
namespace ppforest2::stats {
  namespace simulation::params {
    struct Classification {
      types::Feature mean            = 100.0F; 
      types::Feature mean_separation = 50.0F;  
      types::Feature sd              = 10.0F;  
    };

    struct Regression {
      int n_informative          = 0;    
      types::Feature y_intercept = 0.0F; 
      types::Feature y_sd        = 0.1F; 
      types::Feature sd          = 1.0F; 
    };
  }

  DataPacket simulate(int n, int p, int G, RNG& rng, simulation::params::Classification const& params = {});

  DataPacket simulate(int n, int p, RNG& rng, simulation::params::Regression const& params = {});

  struct Split {
    std::vector<int> tr; 
    std::vector<int> te; 
  };

  Split split(DataPacket const& data, float train_ratio, RNG& rng);
}