treelite
Functions
xgboost.cc File Reference

Frontend for xgboost model. More...

#include <dmlc/data.h>
#include <dmlc/memory_io.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...
 
void ExportXGBoostModel (const char *filename, const Model &model, const char *name_obj)
 export a model in XGBoost format. The exported model can be read by XGBoost (dmlc/xgboost). 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

void ExportXGBoostModel ( const char *  filename,
const Model model,
const char *  name_obj 
)

export a model in XGBoost format. The exported model can be read by XGBoost (dmlc/xgboost).

Parameters
filenamename of model file
modelmodel to export
name_objname of objective function

Definition at line 33 of file xgboost.cc.

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 28 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 39 of file xgboost.cc.