10 #ifndef TREELITE_GTIL_H_ 11 #define TREELITE_GTIL_H_ 33 std::size_t
Predict(
const Model* model,
const DMatrix* input,
float* output,
int nthread,
47 std::size_t
Predict(
const Model* model,
const float* input, std::size_t num_row,
float* output,
48 int nthread,
bool pred_transform);
69 #endif // TREELITE_GTIL_H_
std::size_t Predict(const Model *model, const DMatrix *input, float *output, int nthread, bool pred_transform)
Predict with a DMatrix.
std::size_t GetPredictOutputSize(const Model *model, std::size_t num_row)
Given a batch of data rows, query the necessary size of array to hold predictions for all data points...