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 <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)
 
Model LoadXGBoostModel (const char *filename)
 load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble. More...
 
Model LoadXGBoostModel (const void *buf, size_t len)
 load an XGBoost model from a memory buffer. More...
 

Detailed Description

Frontend for xgboost model.

Copyright 2017 by Contributors

Author
Philip Cho

Definition in file xgboost.cc.

Function Documentation

Model LoadXGBoostModel ( const char *  filename)

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

Parameters
filenamename of model file
Returns
loaded model

Definition at line 27 of file xgboost.cc.

Model LoadXGBoostModel ( const void *  buf,
size_t  len 
)

load an XGBoost model from a memory buffer.

Parameters
bufmemory buffer
lensize of memory buffer
Returns
loaded model

Definition at line 32 of file xgboost.cc.