Types.hpp

Go to the documentation of this file.

#pragma once
 
#include "utils/UserError.hpp"
 
#include <Eigen/Dense>
#include <string>
#include <string_view>
#include <vector>

namespace ppforest2::types {
  using Feature = float;

  using GroupId = int;

  using Outcome = Feature;

  using FeatureMatrix = Eigen::Matrix<Feature, Eigen::Dynamic, Eigen::Dynamic>;

  using FeatureVector = Eigen::Matrix<Feature, Eigen::Dynamic, 1>;

  using GroupIdVector = Eigen::Matrix<GroupId, Eigen::Dynamic, 1>;

  using OutcomeVector = Eigen::Matrix<Outcome, Eigen::Dynamic, 1>;

  using Names = std::vector<std::string>;

  template<typename T> using Matrix = Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic>;

  template<typename T> using Vector = Eigen::Matrix<T, Eigen::Dynamic, 1>;

  enum class Mode : uint8_t { Classification, Regression };

  inline bool is_classification(Mode mode) {
    return mode == Mode::Classification;
  }

  inline bool is_regression(Mode mode) {
    return mode == Mode::Regression;
  }

  inline std::string to_string(Mode mode) {
    return is_regression(mode) ? "regression" : "classification";
  }

  inline Mode mode_from_string(std::string_view s) {
    if (s == "classification") {
      return Mode::Classification;
    }
 
    if (s == "regression") {
      return Mode::Regression;
    }
 
    throw UserError("Invalid mode '" + std::string(s) + "'. Expected 'classification' or 'regression'.");
  }
 
}