DataPacket.hpp

Go to the documentation of this file.

#pragma once
 
#include "stats/Stats.hpp"
#include "utils/Types.hpp"
 
#include <set>
#include <string>
#include <vector>
 
namespace ppforest2::stats {
  struct DataPacket {
    types::FeatureMatrix const x;
    types::Vector<types::Response> const y;
    std::set<types::Response> const groups;
    std::vector<std::string> const group_names;
    std::vector<std::string> const feature_names;
 
    DataPacket(types::FeatureMatrix const& x,
               types::Vector<types::Response> const& y,
               std::set<types::Response> const& groups,
               std::vector<std::string> const& group_names   = {},
               std::vector<std::string> const& feature_names = {})
        : x(x)
        , y(y)
        , groups(groups)
        , group_names(group_names)
        , feature_names(feature_names) {}
 
    DataPacket(types::FeatureMatrix const& x,
               types::Vector<types::Response> const& y,
               std::vector<std::string> const& group_names   = {},
               std::vector<std::string> const& feature_names = {})
        : x(x)
        , y(y)
        , groups(unique(y))
        , group_names(group_names)
        , feature_names(feature_names) {}
 
    DataPacket()
        : x(types::FeatureMatrix())
        , y(types::Vector<types::Response>())
        , groups(std::set<types::Response>()) {}
  };
}