Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dmlc/treelite
/ types & classes
Types & classes
133 in github.com/dmlc/treelite
⨍
Functions
686
◇
Types & classes
133
↳
Endpoints
4
↓ 14 callers
Class
Model
Decision tree ensemble model Parameters ---------- handle : Handle to C++ object
python/treelite/model.py:16
↓ 11 callers
Class
ArrayOfArrays
Utility class to marshall a collection of arrays in order to pass to a C function
python/treelite/sklearn/importer.py:15
↓ 11 callers
Class
TreeliteError
Error thrown by Treelite
python/treelite/core.py:12
↓ 7 callers
Class
Metadata
Metadata object, consisting of metadata information about the model at large. Parameters ---------- num_feature: Number of f
python/treelite/model_builder.py:14
↓ 7 callers
Class
ModelBuilder
Model builder class, to iteratively build a tree ensemble model. .. note:: The model builder object must be only accessed by a sing
python/treelite/model_builder.py:115
↓ 7 callers
Class
PostProcessorFunc
Specification for postprocessor of prediction outputs Parameters ---------- name: Name of the postprocessor. Consult :py:doc
python/treelite/model_builder.py:81
↓ 7 callers
Class
TreeAnnotation
Annotation for individual trees. Use this object to look up which target and class each tree is associated with. The output of each targ
python/treelite/model_builder.py:49
↓ 3 callers
Class
GTILConfig
Object holding configuration data
python/treelite/gtil/gtil.py:19
↓ 3 callers
Class
LogMessageFatal
include/treelite/logging.h:115
↓ 3 callers
Class
_TreelitePyBufferFrame
Abridged buffer structure used by Treelite
python/treelite/model.py:331
↓ 2 callers
Class
ThreadConfig
! * \brief Represent thread configuration, to be used with parallel loops. */
include/treelite/detail/threading_utils.h:66
↓ 2 callers
Class
_PyBuffer
The full Python buffer structure as defined by PEP 3118
python/treelite/model.py:342
↓ 1 callers
Class
BuildConfiguration
Configurations use when building libtreelite
python/packager/build_config.py:8
↓ 1 callers
Class
DenseMatrixAccessor
src/gtil/predict.cc:43
↓ 1 callers
Class
Error
! * \brief Exception class that will be thrown by Treelite */
include/treelite/error.h:18
↓ 1 callers
Class
HeaderAccessor
Accessor for fields in the header Parameters ---------- model: The model object
python/treelite/model.py:433
↓ 1 callers
Class
LogMessage
include/treelite/logging.h:135
↓ 1 callers
Class
LogMessageWarning
include/treelite/logging.h:152
↓ 1 callers
Class
SparseMatrixAccessor
src/gtil/predict.cc:59
↓ 1 callers
Class
TreeAccessor
Accessor for fields in a tree Parameters ---------- model: The model object tree_id: ID of the tree
python/treelite/model.py:503
↓ 1 callers
Class
TreeliteLibraryNotFound
Error thrown by when Treelite is not found
python/treelite/libpath.py:11
Class
APIErrorEntry
src/c_api/c_api_error.cc:16
Class
ArrayHandler
src/model_loader/detail/xgboost_json/delegated_handler.h:297
Class
BaseHandler
src/model_loader/detail/xgboost_json/delegated_handler.h:95
Class
BufferSerializerMixIn
! * \brief Mixin that writes directly to a pre-allocated buffer. * * This avoids the overhead of std::ostringstream which has poor performance * d
include/treelite/detail/serializer_mixins.h:166
Class
CategoryContainerHandler
! \brief Handler for CategoryContainer objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:462
Class
CategoryInfoArrayHandler
! \brief Handler for array of objects of CategoryInfo type */
src/model_loader/detail/xgboost_json/delegated_handler.h:472
Class
CategoryInfoHandler
! \brief Handler for CategoryInfo objects */
src/model_loader/detail/xgboost_json/delegated_handler.h:479
Class
Configuration
! \brief Configuration class */
include/treelite/gtil.h:50
Class
ContiguousArray
include/treelite/contiguous_array.h:17
Class
CustomBuildHook
A custom build hook
python/hatch_build.py:18
Class
DelegatedHandler
src/model_loader/detail/xgboost_json/sax_adapters.h:23
Class
DelegatedHandler
! \brief Handler which delegates JSON parsing to stack of delegates */
src/model_loader/detail/xgboost_json/delegated_handler.h:212
Class
Delegator
! \brief Class for handling delegation of JSON handling */
src/model_loader/detail/xgboost_json/delegated_handler.h:99
Class
Deserializer
src/serializer.cc:182
Class
Deserializer
include/treelite/tree.h:53
Enum
FeatureType
src/model_loader/detail/xgboost.h:34
Class
GBTreeModelHandler
! \brief Handler for GBTreeModel objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:447
Class
GBTreeModelParam
src/model_loader/xgboost_legacy.cc:169
Class
GradientBoosterHandler
! \brief Handler for GradientBoosterHandler objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:491
Class
GradientBoostingBinaryClassifierMixIn
src/model_loader/sklearn.cc:82
Class
GradientBoostingMulticlassClassifierMixIn
src/model_loader/sklearn.cc:106
Class
GradientBoostingRegressorMixIn
src/model_loader/sklearn.cc:59
Class
HandlerConfig
src/model_loader/detail/xgboost_json/delegated_handler.h:46
Class
HistGradientBoostingBinaryClassifierMixIn
src/model_loader/sklearn.cc:158
Class
HistGradientBoostingMulticlassClassifierMixIn
src/model_loader/sklearn.cc:176
Class
HistGradientBoostingNode
src/model_loader/sklearn.cc:262
Class
HistGradientBoostingNode<std::int32_t>
src/model_loader/sklearn.cc:277
Class
HistGradientBoostingNode<std::int64_t>
src/model_loader/sklearn.cc:278
Class
HistGradientBoostingRegressorMixIn
src/model_loader/sklearn.cc:135
Class
IgnoreHandler
src/model_loader/detail/xgboost_json/delegated_handler.cc:44
Class
IgnoreHandler
! \brief JSON handler that ignores all delegated input */
src/model_loader/detail/xgboost_json/delegated_handler.h:263
Class
IsolationForestMixIn
src/model_loader/sklearn.cc:33
Class
LGBTree
src/model_loader/lightgbm.cc:186
Class
LearnerHandler
! \brief Handler for Learner objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:529
Class
LearnerModelParam
src/model_loader/xgboost_legacy.cc:154
Class
LearnerParamHandler
! \brief Handler for LearnerParam objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:519
Class
LogCallbackRegistry
include/treelite/logging.h:169
Enum
Masks
src/model_loader/lightgbm.cc:182
Class
Metadata
include/treelite/model_builder.h:31
Enum
MissingType
src/model_loader/lightgbm.cc:184
Class
Model
src/gtil/postprocessor.h:16
Class
Model
include/treelite/gtil.h:21
Class
Model
include/treelite/model_loader.h:18
Class
Model
include/treelite/model_builder.h:27
Class
Model
! \brief Model class for tree ensemble model */
include/treelite/tree.h:454
Class
ModelBuilder
! * \brief Model builder interface. * \note A model builder object must be accessed by a single thread. For parallel tree construction, * bui
include/treelite/model_builder.h:40
Class
ModelBuilderImpl
src/model_builder/model_builder.cc:59
Enum
ModelBuilderState
src/model_builder/model_builder.cc:50
Class
ModelParamV3
src/serializer.cc:44
Class
ModelPreset
include/treelite/tree.h:44
Class
NlohmannJSONAdapter
! * \brief Adapter for SAX parser from nlohmann/json */
src/model_loader/detail/xgboost_json/sax_adapters.h:54
Class
Node
src/model_loader/xgboost_legacy.cc:199
Class
NodeStat
src/model_loader/xgboost_legacy.cc:190
Class
NodeV3
src/serializer.cc:52
Class
OMPException
! * \brief OMP Exception class catches, saves and rethrows exception from OMP blocks */
include/treelite/detail/omp_exception.h:20
Class
ObjectMemberHandler
src/model_builder/detail/json_parsing.h:142
Class
ObjectiveHandler
! \brief Handler for ObjectiveHandler objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:508
Enum
Operator
! \brief Type of comparison operators used in numerical test nodes */
include/treelite/enum/operator.h:17
Class
OutputHandler
src/model_loader/detail/xgboost_json/delegated_handler.h:279
Enum
OutputType
src/serializer.cc:37
Class
ParallelSchedule
OpenMP schedule
include/treelite/detail/threading_utils.h:87
Class
ParametrizedTestSuite
tests/cpp/test_gtil.cc:31
Class
ParsedCategoryContainer
src/model_loader/detail/xgboost_json/delegated_handler.h:65
Class
ParsedCategoryInfo
src/model_loader/detail/xgboost_json/delegated_handler.h:59
Class
ParsedLearnerParams
src/model_loader/detail/xgboost_json/delegated_handler.h:87
Class
ParsedRegTreeParams
src/model_loader/detail/xgboost_json/delegated_handler.h:82
Class
ParsedStringCategorical
src/model_loader/detail/xgboost_json/delegated_handler.h:51
Class
ParsedXGBoostModel
src/model_loader/detail/xgboost_json/delegated_handler.h:71
Class
PeekableInputStream
peekable input stream implemented with a ring buffer */
src/model_loader/xgboost_legacy.cc:60
Class
PostProcessorFunc
include/treelite/model_builder.h:33
Enum
PredictKind
! \brief Prediction type */
include/treelite/gtil.h:26
Class
ProbToMargin
src/model_loader/detail/xgboost.h:16
Class
PyBufferDeserializerMixIn
include/treelite/detail/serializer_mixins.h:98
Class
PyBufferSerializerMixIn
include/treelite/detail/serializer_mixins.h:72
Class
RapidJSONAdapter
! * \brief Adapter for SAX parser from RapidJSON */
src/model_loader/detail/xgboost_json/sax_adapters.h:28
Class
RegTreeArrayHandler
! \brief Handler for array of objects of Tree type */
src/model_loader/detail/xgboost_json/delegated_handler.h:433
Class
RegTreeHandler
! \brief Handler for RegTree objects from XGBoost schema */
src/model_loader/detail/xgboost_json/delegated_handler.h:398
Class
ReturnValueEntry
! \brief When returning a complex object from a C API function, we * store the object here and then return a pointer. The * storage
src/c_api/c_api_utils.h:22
Class
RootHandler
src/model_loader/detail/xgboost_json/delegated_handler.h:96
next →
1–100 of 133, ranked by callers