ML functions
 
Loading...
Searching...
No Matches
XGBoost.h File Reference

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< XGBoostml::XGBoostPtr
 Forward declaration of the XGBoost class.
 

Detailed Description

This file contains the implementation of XGBoost-based machine learning functions in Velox.

Typedef Documentation

◆ XGBoostPtr

typedef std::shared_ptr<XGBoost> ml::XGBoostPtr

Forward declaration of the XGBoost class.

Alias for a shared pointer to an XGBoost object.