include/XGBoostPredictor.h

#ifndef HZZ2L2NU_INCLUDE_XGBOOSTPREDICTOR_H_
#define HZZ2L2NU_INCLUDE_XGBOOSTPREDICTOR_H_

#include <exception>
#include <string>

#include <xgboost/c_api.h>


class XGBoostPredictorException : public std::exception {
 public:
  XGBoostPredictorException(std::string const &functionName, int exitCode);

  char const *what() const noexcept override {
    return message_.c_str();
  }

 private:
  std::string message_;
};


class XGBoostPredictor {
 public:
  XGBoostPredictor(std::string const &path, int numFeatures);

  ~XGBoostPredictor();

  float Predict(float const *x) const;

 private:
  static void CheckCall(std::string const &functionName, int exitCode);

  BoosterHandle booster_;

  int numFeatures_;
};

#endif  // HZZ2L2NU_INCLUDE_XGBOOSTPREDICTOR_H_

---