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 {
  struct SimulationParams {
    float mean            = 100.0f; 
    float mean_separation = 50.0f;  
    float sd              = 10.0f;  
  };

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

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

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