gtil.h

Go to the documentation of this file.

 
#ifndef TREELITE_GTIL_H_
#define TREELITE_GTIL_H_
 
#include <cstddef>
#include <cstdint>
#include <optional>
#include <string>
#include <variant>
#include <vector>
 
namespace treelite {
 
class Model;
 
namespace gtil {
 
enum class PredictKind : std::int8_t {
  kPredictDefault = 0,
  kPredictRaw = 1,
  kPredictLeafID = 2,
  kPredictPerTree = 3
};
 
struct Configuration {
  int nthread{0};  // use all threads by default
  PredictKind pred_kind{PredictKind::kPredictDefault};
  Configuration() = default;
  explicit Configuration(std::string const& config_json);
};
 
template <typename InputT>
void Predict(Model const& model, InputT const* input, std::uint64_t num_row, InputT* output,
    Configuration const& config);
 
template <typename InputT>
void PredictSparse(Model const& model, InputT const* data, std::uint64_t const* col_ind,
    std::uint64_t const* row_ptr, std::uint64_t num_row, InputT* output,
    Configuration const& config);
 
std::vector<std::uint64_t> GetOutputShape(
    Model const& model, std::uint64_t num_row, Configuration const& config);
 
extern template void Predict<float>(
    Model const&, float const*, std::uint64_t, float*, Configuration const&);
extern template void Predict<double>(
    Model const&, double const*, std::uint64_t, double*, Configuration const&);
extern template void PredictSparse<float>(Model const&, float const*, std::uint64_t const*,
    std::uint64_t const*, std::uint64_t, float*, Configuration const&);
extern template void PredictSparse<double>(Model const&, double const*, std::uint64_t const*,
    std::uint64_t const*, std::uint64_t, double*, Configuration const&);
 
}  // namespace gtil
}  // namespace treelite
 
#endif  // TREELITE_GTIL_H_