Treelite
|
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>
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... | |
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.
filename | name of model file |
out | reference to loaded model |
Definition at line 28 of file xgboost.cc.
void LoadXGBoostModel | ( | const void * | buf, |
size_t | len, | ||
Model * | out | ||
) |
load an XGBoost model from a memory buffer.
buf | memory buffer |
len | size of memory buffer |
out | reference to loaded model |
Definition at line 33 of file xgboost.cc.