treelite
|
Frontend for xgboost model. More...
#include <dmlc/data.h>
#include <dmlc/memory_io.h>
#include <treelite/tree.h>
#include <memory>
#include <queue>
#include <cstring>
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... | |
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).
filename | name of model file |
model | model to export |
name_obj | name 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.
filename | name of model file |
Definition at line 28 of file xgboost.cc.
Model LoadXGBoostModel | ( | const void * | buf, |
size_t | len | ||
) |
load an XGBoost model from a memory buffer.
buf | memory buffer |
len | size of memory buffer |
Definition at line 39 of file xgboost.cc.