Treelite
Functions
xgboost.cc File Reference

Frontend for xgboost model. More...

#include <dmlc/data.h>
#include <dmlc/memory_io.h>
#include <treelite/frontend.h>
#include <treelite/tree.h>
#include <algorithm>
#include <memory>
#include <queue>
#include <cstring>
Include dependency graph for xgboost.cc:

Go to the source code of this file.

Functions

 DMLC_REGISTRY_FILE_TAG (xgboost)
 
void LoadXGBoostModel (const char *filename, Model *out)
 load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble. More...
 
void LoadXGBoostModel (const void *buf, size_t len, Model *out)
 load an XGBoost model from a memory buffer. More...
 

Detailed Description

Frontend for xgboost model.

Copyright (c) 2017-2020 by Contributors

Author
Hyunsu Cho

Definition in file xgboost.cc.

Function Documentation

◆ LoadXGBoostModel() [1/2]

void LoadXGBoostModel ( const char *  filename,
Model out 
)

load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble.

Parameters
filenamename of model file
outreference to loaded model

Definition at line 28 of file xgboost.cc.

◆ LoadXGBoostModel() [2/2]

void LoadXGBoostModel ( const void *  buf,
size_t  len,
Model out 
)

load an XGBoost model from a memory buffer.

Parameters
bufmemory buffer
lensize of memory buffer
outreference to loaded model

Definition at line 33 of file xgboost.cc.