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 x;
    types::OutcomeVector y;
    std::set<types::GroupId> groups;
    types::Names group_names;
    types::Names feature_names;
 
    DataPacket(
        types::FeatureMatrix const& x,
        types::OutcomeVector const& y,
        std::set<types::GroupId> const& groups,
        types::Names const& group_names   = {},
        types::Names const& feature_names = {}
    )
        : x(x)
        , y(y)
        , groups(groups)
        , group_names(group_names)
        , feature_names(feature_names) {}
 
    DataPacket(
        types::FeatureMatrix const& x,
        types::OutcomeVector const& y,
        types::Names const& group_names   = {},
        types::Names const& feature_names = {}
    )
        : x(x)
        , y(y)
        , groups(unique(y.cast<types::GroupId>()))
        , group_names(group_names)
        , feature_names(feature_names) {}
 
    DataPacket() = default;
  };
}