Program Listing for File stats_utils.hpp

↰ Return to documentation for file (include/uitsl/math/stats_utils.hpp)

#pragma once
#ifndef UITSL_MATH_STATS_UTILS_HPP_INCLUDE
#define UITSL_MATH_STATS_UTILS_HPP_INCLUDE

#include <stddef.h>

#include "../../../third-party/Empirical/source/base/vector.h"
#include "../../../third-party/Empirical/source/data/DataNode.h"

namespace uitsl {

double sample_mean_with_replacement(
  emp::Random & rand,
  const emp::vector<double>& data
) {

  emp::DataNode<double, emp::data::Range> sampled;

  for (size_t sample = 0; sample < data.size(); ++sample) {
    sampled.Add(
      data[rand.GetUInt(data.size())]
    );
  }

  return sampled.GetMean();

}


std::tuple<double, double> bootstrap(
  emp::Random & rand,
  const emp::vector<double>& data,
  const double percentile=5.0,
  const size_t num_reps=10000
) {

  emp_assert(percentile <= 50.0);

  emp::DataNode<double, emp::data::Log> sampled;

  for (size_t rep = 0; rep < num_reps; ++rep) {
    sampled.Add(
      sample_mean_with_replacement(rand, data)
    );
  }

  return {
    sampled.GetPercentile(percentile),
    sampled.GetPercentile(100.0 - percentile)
  };

}

} // namespace uitsl

#endif // #ifndef UITSL_MATH_STATS_UTILS_HPP_INCLUDE