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 | |
void | LoadLightGBMModel (const char *filename, Model *out) |
load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble. 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. More... | |
void | LoadXGBoostModel (const void *buf, size_t len, Model *out) |
load an XGBoost model from a memory buffer. More... | |
void | LoadProtobufModel (const char *filename, Model *out) |
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 (c) 2017-2020 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 344 of file protobuf.cc.
void LoadLightGBMModel | ( | const char * | filename, |
Model * | out | ||
) |
load a model file generated by LightGBM (Microsoft/LightGBM). The model file must contain a decision tree ensemble.
filename | name of model file |
out | reference to loaded model |
Definition at line 26 of file lightgbm.cc.
void LoadProtobufModel | ( | const char * | filename, |
Model * | out | ||
) |
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 |
out | reference to loaded model |
Definition at line 340 of file protobuf.cc.
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.