This file contains the implementation of XGBoost-based machine learning functions in Velox. More...
#include <dirent.h>#include <fcntl.h>#include <stdlib.h>#include <unistd.h>#include <xgboost/c_api.h>#include <cmath>#include <iostream>#include <memory>#include <string>#include "BaseFunction.h"#include "velox/exec/tests/utils/AssertQueryBuilder.h"#include "velox/exec/tests/utils/PlanBuilder.h"#include "velox/exec/tests/utils/TempDirectoryPath.h"#include "velox/ml_functions/DecisionTree.h"#include "velox/vector/tests/utils/VectorTestBase.h"Go to the source code of this file.
Classes | |
| class | ml::XGBoost |
| A class for managing XGBoost models and making predictions. More... | |
| class | ml::XGBoostPrediction |
| A machine learning function for making predictions using XGBoost models. More... | |
Typedefs | |
| typedef std::shared_ptr< XGBoost > | ml::XGBoostPtr |
| Forward declaration of the XGBoost class. | |
This file contains the implementation of XGBoost-based machine learning functions in Velox.
| typedef std::shared_ptr<XGBoost> ml::XGBoostPtr |
Forward declaration of the XGBoost class.
Alias for a shared pointer to an XGBoost object.