treelite
|
Collection of front-end methods to load or construct ensemble model. More...
Go to the source code of this file.
Classes | |
class | TreeBuilder |
tree builder class More... | |
class | ModelBuilder |
model builder class More... | |
Functions | |
Model | LoadLightGBMModel (const char *filename) |
load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble. More... | |
Model | LoadXGBoostModel (const char *filename) |
load a model file generated by XGBoost (dmlc/xgboost). The model file must contain a decision tree ensemble. More... | |
Model | LoadXGBoostModel (const void *buf, size_t len) |
load an XGBoost model from a memory buffer. More... | |
Model | LoadProtobufModel (const char *filename) |
load a model in Protocol Buffers format. Protocol Buffers (google/protobuf) is a language- and platform-neutral mechanism for serializing structured data. See tree.proto for format spec. More... | |
void | ExportProtobufModel (const char *filename, const Model &model) |
export a model in Protocol Buffers format. Protocol Buffers (google/protobuf) is a language- and platform-neutral mechanism for serializing structured data. See src/tree.proto for format spec. More... | |
Collection of front-end methods to load or construct ensemble model.
Copyright 2017 by Contributors
Definition in file frontend.h.
void ExportProtobufModel | ( | const char * | filename, |
const Model & | model | ||
) |
export a model in Protocol Buffers format. Protocol Buffers (google/protobuf) is a language- and platform-neutral mechanism for serializing structured data. See src/tree.proto for format spec.
filename | name of model file |
model | model to export |
Definition at line 341 of file protobuf.cc.
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 25 of file lightgbm.cc.
Model LoadProtobufModel | ( | const char * | filename | ) |
load a model in Protocol Buffers format. Protocol Buffers (google/protobuf) is a language- and platform-neutral mechanism for serializing structured data. See tree.proto for format spec.
filename | name of model file |
Definition at line 336 of file protobuf.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 33 of file xgboost.cc.