treelite
|
▼ include | |
▼ treelite | |
annotator.h | Branch annotation tools |
base.h | Defines configuration macros of treelite |
c_api.h | C API of treelite, used for interfacing with other languages This header is excluded from the runtime |
c_api_common.h | |
common.h | |
compiler.h | Interface of compiler that compiles a tree ensemble model |
data.h | Input data structure of treelite |
frontend.h | Collection of front-end methods to load or construct ensemble model |
logging.h | |
omp.h | Compatiblity wrapper for systems that don't support OpenMP |
tree.h | Model structure for tree |
▼ runtime | |
▼ native | |
▼ include | |
▼ treelite | |
c_api_common.h | |
c_api_runtime.h | C API of treelite, used for interfacing with other languages This header is used exclusively by the runtime |
common.h | |
entry.h | |
logging.h | |
predictor.h | Load prediction function exported as a shared library |
▼ src | |
▼ c_api | |
c_api_common.cc | |
c_api_error.cc | |
c_api_error.h | |
c_api_runtime.cc | C API of treelite (runtime portion) |
▼ common | |
filesystem.h | |
math.h | |
▼ thread_pool | |
spsc_queue.h | Lock-free single-producer-single-consumer queue |
thread_pool.h | Simple thread pool implementation |
logging.cc | |
predictor.cc | Load prediction function exported as a shared library |
▼ src | |
▼ c_api | |
c_api.cc | C API of treelite, used for interfacing with other languages |
c_api_common.cc | |
c_api_error.cc | |
c_api_error.h | |
optable.cc | Conversion table from string to Operator |
▼ common | |
filesystem.h | |
math.h | |
▼ compiler | |
▼ ast | |
ast.h | Definition for AST classes |
build.cc | Build AST from a given model |
builder.h | AST Builder class |
dump.cc | Generate text representation of AST |
fold_code.cc | |
is_categorical_array.cc | AST manipulation logic to determine whether each feature is categorical or not |
load_data_counts.cc | AST manipulation logic to load data counts |
quantize.cc | Quantize thresholds in condition nodes |
split.cc | Split prediction subroutine into multiple translation units (files) |
▼ common | |
categorical_bitmap.h | Function to generate bitmaps for categorical splits |
code_folding_util.h | Utilities for code folding |
▼ elf | |
elf_formatter.cc | Generate a relocatable object file containing a constant, read-only array |
elf_formatter.h | Generate a relocatable object file containing a constant, read-only array |
▼ native | |
code_folder_template.h | Template for evaluation logic for folded code |
header_template.h | Template for header |
main_template.h | Template for main function |
pred_transform.h | |
qnode_template.h | Code template for QuantizerNode |
ast_native.cc | C code generator |
compiler.cc | Registry of compilers |
failsafe.cc | C code generator (fail-safe). The generated code will mimic prediction logic found in XGBoost |
param.h | Parameters for tree compiler |
pred_transform.cc | Library of transform functions to convert margins into predictions |
pred_transform.h | |
▼ frontend | |
builder.cc | Model builder frontend |
lightgbm.cc | Frontend for lightgbm model |
model.cc | Placeholder to register ModelParam with the DMLC parameter registry |
protobuf.cc | Frontend for protobuf model |
xgboost.cc | Frontend for xgboost model |
annotator.cc | Branch annotation tools |
data.cc | |
logging.cc |