A class for managing XGBoost models and making predictions.
More...
#include <XGBoost.h>
|
| | XGBoost (std::string pathToJSON) |
| | Construct a new XGBoost object.
|
| |
| void | predict (VectorPtr &input, std::vector< float > &resultVector, int numInputs, int numFeatures) |
| | Make predictions using the XGBoost model.
|
| |
|
|
BoosterHandle | booster |
| | Handle to the XGBoost booster.
|
| |
A class for managing XGBoost models and making predictions.
◆ XGBoost()
| ml::XGBoost::XGBoost |
( |
std::string | pathToJSON | ) |
|
|
inline |
Construct a new XGBoost object.
- Parameters
-
| pathToJSON | The path to the JSON file containing the XGBoost model. |
◆ predict()
| void ml::XGBoost::predict |
( |
VectorPtr & | input, |
|
|
std::vector< float > & | resultVector, |
|
|
int | numInputs, |
|
|
int | numFeatures ) |
|
inline |
Make predictions using the XGBoost model.
- Parameters
-
| input | The input vector containing the features. |
| resultVector | The output vector to store the predictions. |
| numInputs | The number of input rows. |
| numFeatures | The number of features per input row. |
The documentation for this class was generated from the following file: