Treelite
|
Collection of front-end methods to load or construct ensemble model. More...
#include <treelite/base.h>
#include <string>
#include <memory>
#include <vector>
#include <cstdint>
#include "frontend_impl.h"
Go to the source code of this file.
Classes | |
class | Value |
class | TreeBuilder |
tree builder class More... | |
class | ModelBuilder |
model builder class More... | |
Functions | |
std::unique_ptr< treelite::Model > | LoadLightGBMModel (const char *filename) |
load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble. 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. 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 > | LoadXGBoostJSONModel (const char *filename) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble in the JSON format. More... | |
std::unique_ptr< treelite::Model > | LoadXGBoostJSONModelString (const char *json_str, size_t length) |
load an XGBoost model from a JSON string More... | |
Collection of front-end methods to load or construct ensemble model.
Copyright (c) 2017-2020 by Contributors
Definition in file frontend.h.
std::unique_ptr< treelite::Model > LoadLightGBMModel | ( | const char * | filename | ) |
load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble.
filename | name of model file |
Definition at line 26 of file lightgbm.cc.
std::unique_ptr< treelite::Model > LoadXGBoostJSONModel | ( | const char * | filename | ) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble in the JSON format.
filename | name of model file |
out | reference to loaded model |
Definition at line 45 of file xgboost_json.cc.
std::unique_ptr< treelite::Model > LoadXGBoostJSONModelString | ( | const char * | json_str, |
size_t | length | ||
) |
load an XGBoost model from a JSON string
json_str | JSON char array |
length | length of JSON char array |
out | reference to loaded model |
Definition at line 85 of file xgboost_json.cc.
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.