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 | C API of Treelite, used for interfacing with other languages This header is used by both the runtime and the main package |
c_api_error.h | Error handling for C API |
c_api_runtime.h | C API of Treelite, used for interfacing with other languages This header is used exclusively by the runtime |
compiler.h | Interface of compiler that compiles a tree ensemble model |
compiler_param.h | Parameters for tree compiler |
data.h | Input data structure of Treelite |
filesystem.h | Cross-platform wrapper for common filesystem functions |
frontend.h | Collection of front-end methods to load or construct ensemble model |
frontend_impl.h | Implementation for frontend.h |
logging.h | Logging facility for Treelite |
math.h | Some useful math utilities |
omp.h | Compatiblity wrapper for systems that don't support OpenMP |
predictor.h | Load prediction function exported as a shared library |
tree.h | Model structure for tree ensemble |
tree_impl.h | Implementation for tree.h |
typeinfo.h | Defines TypeInfo class and utilities |
▼ src | |
▼ c_api | |
c_api.cc | C API of treelite, used for interfacing with other languages |
c_api_common.cc | C API of treelite (this file is used by both runtime and main package) |
c_api_error.cc | C error handling |
c_api_runtime.cc | C API of treelite (runtime portion) |
▼ 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 |
format_util.h | Formatting utilities |
▼ 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 |
typeinfo_ctypes.h | Look up C symbols corresponding to TypeInfo |
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 |
pred_transform.cc | Library of transform functions to convert margins into predictions |
pred_transform.h | |
▼ frontend | |
▼ xgboost | |
xgboost.h | Helper functions for loading XGBoost models |
xgboost_json.h | Methods for loading models from XGBoost-style JSON |
builder.cc | Model builder frontend |
lightgbm.cc | Frontend for LightGBM model |
xgboost.cc | Frontend for xgboost model |
xgboost_json.cc | Frontend for xgboost model |
xgboost_util.cc | Common utilities for XGBoost frontends |
▼ predictor | |
▼ thread_pool | |
spsc_queue.h | Lock-free single-producer-single-consumer queue |
thread_pool.h | Simple thread pool implementation |
predictor.cc | Load prediction function exported as a shared library |
annotator.cc | Branch annotation tools |
data.cc | Input data structure of Treelite |
filesystem.cc | Cross-platform wrapper for common filesystem functions |
logging.cc | Logging facility for treelite |
optable.cc | Conversion tables to obtain Operator from string |
reference_serializer.cc | Reference serializer implementation |
typeinfo.cc | Conversion tables to obtain TypeInfo from string |