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 |
error.h | Exception class used throughout the Treelite codebase |
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 |
gtil.h | General Tree Inference Library (GTIL), providing a reference implementation for predicting with decision trees. GTIL is useful in cases it is infeasible to build the tree models as native shared libs |
logging.h | Logging facility for Treelite |
math.h | Some useful math utilities |
omp.h | Compatiblity wrapper for systems that don't support OpenMP |
omp_exception.h | Utility to propagate exceptions throws inside an OpenMP block |
optional.h | Backport of std::optional from C++17 |
predictor.h | Load prediction function exported as a shared library |
thread_local.h | Helper class for thread-local storage |
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 |
ast_native.h | 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 |
failsafe.h | C code generator (fail-safe). The generated code will mimic prediction logic found in XGBoost |
pred_transform.cc | |
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 |
sklearn.cc | Frontend for scikit-learn models |
xgboost.cc | Frontend for xgboost model |
xgboost_json.cc | Frontend for xgboost model |
xgboost_util.cc | Common utilities for XGBoost frontends |
▼ gtil | |
pred_transform.cc | |
pred_transform.h | |
predict.cc | General Tree Inference Library (GTIL), providing a reference implementation for predicting with decision trees. GTIL is useful in cases it is infeasible to build the tree models as native shared libs |
▼ 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 |
▼ threading_utils | |
parallel_for.h | Implemenation of parallel for loop |
annotator.cc | Branch annotation tools |
data.cc | Input data structure of Treelite |
filesystem.cc | Cross-platform wrapper for common filesystem functions |
json_serializer.cc | Reference serializer implementation, which serializes to JSON. This is useful for testing correctness of the binary serializer |
logging.cc | Logging facility for treelite |
model_concat.cc | Implementation of model concatenation |
optable.cc | Conversion tables to obtain Operator from string |
serializer.cc | Implementation for serialization related functions |
typeinfo.cc | Conversion tables to obtain TypeInfo from string |