Treelite
|
Frontend for xgboost model. More...
#include "xgboost/xgboost.h"
#include <treelite/frontend.h>
#include <treelite/tree.h>
#include <treelite/logging.h>
#include <algorithm>
#include <memory>
#include <queue>
#include <fstream>
#include <sstream>
#include <cstring>
#include <cstdint>
Go to the source code of this file.
Functions | |
std::unique_ptr< treelite::Model > | LoadXGBoostModel (const char *filename) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble. More... | |
std::unique_ptr< treelite::Model > | LoadXGBoostModel (const void *buf, size_t len) |
load an XGBoost model from a memory buffer. More... | |
std::unique_ptr< treelite::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 29 of file xgboost.cc.
std::unique_ptr< treelite::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 34 of file xgboost.cc.