Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dmlc/treelite
/ functions
Functions
686 in github.com/dmlc/treelite
⨍
Functions
686
◇
Types & classes
133
↳
Endpoints
4
↓ 1 callers
Function
BinarySearch
src/model_loader/detail/xgboost_json/delegated_handler.cc:31
↓ 1 callers
Function
BitsetToList
src/model_loader/lightgbm.cc:210
↓ 1 callers
Function
BuildModel
tests/example_app/example.c:22
↓ 1 callers
Function
CanonicalObjective
! * \brief Canonicalize the name of an objective function. * * Some objective functions have many aliases. We use the canonical name to avoid confu
src/model_loader/detail/lightgbm.h:26
↓ 1 callers
Function
ConfigurePostProcessor
src/model_builder/model_builder.cc:34
↓ 1 callers
Method
DefaultChild
! * \brief Index of the node's "default" child, used when feature is missing * \param nid ID of node being queried */
include/treelite/tree.h:183
↓ 1 callers
Function
DetectXGBoostFormat
src/model_loader/detail/xgboost.cc:83
↓ 1 callers
Function
DumpTreeAsJSON
src/json_serializer.cc:164
↓ 1 callers
Method
GetCallbackLogInfo
include/treelite/logging.h:178
↓ 1 callers
Method
GetCallbackLogWarning
include/treelite/logging.h:184
↓ 1 callers
Function
GetDepth
src/model_query.cc:18
↓ 1 callers
Method
GetFrames
include/treelite/detail/serializer_mixins.h:90
↓ 1 callers
Method
GetHeaderField
src/field_accessor.cc:16
↓ 1 callers
Function
GetMissingType
src/model_loader/lightgbm.cc:206
↓ 1 callers
Method
GetTotalSize
include/treelite/detail/serializer_mixins.h:152
↓ 1 callers
Method
GetTreeDepth
src/model_query.cc:43
↓ 1 callers
Method
GetTreeField
src/field_accessor.cc:107
↓ 1 callers
Function
GetTreeFieldImpl
src/field_accessor.cc:126
↓ 1 callers
Method
HandleLeafNode
src/model_loader/sklearn.cc:50
↓ 1 callers
Method
HasDataCount
! * \brief Test whether this node has data count * \param nid ID of node being queried */
include/treelite/tree.h:282
↓ 1 callers
Method
HasGain
! * \brief Test whether this node has gain value * \param nid ID of node being queried */
include/treelite/tree.h:311
↓ 1 callers
Method
HasSumHess
! * \brief Test whether this node has hessian sum * \param nid ID of node being queried */
include/treelite/tree.h:297
↓ 1 callers
Method
Init
src/model_loader/xgboost_legacy.cc:282
↓ 1 callers
Function
InitArrayFromPyBuffer
include/treelite/detail/serializer.h:113
↓ 1 callers
Method
Load
src/model_loader/xgboost_legacy.cc:288
↓ 1 callers
Function
LoadGradientBoostingClassifier
src/model_loader/sklearn.cc:395
↓ 1 callers
Function
LoadGradientBoostingRegressor
src/model_loader/sklearn.cc:385
↓ 1 callers
Function
LoadHistGradientBoostingClassifier
src/model_loader/sklearn.cc:428
↓ 1 callers
Function
LoadHistGradientBoostingRegressor
src/model_loader/sklearn.cc:417
↓ 1 callers
Function
LoadIsolationForest
src/model_loader/sklearn.cc:373
↓ 1 callers
Function
LoadLightGBMModel
src/model_loader/lightgbm.cc:40
↓ 1 callers
Function
LoadLightGBMModelFromString
src/model_loader/lightgbm.cc:45
↓ 1 callers
Function
LoadRandomForestClassifier
* Load a RandomForestClassifier using bulk construction * * This is an optimized version that constructs trees in bulk rather than * going through
src/model_loader/sklearn_bulk.cc:232
↓ 1 callers
Function
LoadRandomForestRegressor
* Load a RandomForestRegressor using bulk construction * * This is an optimized version that constructs trees in bulk rather than * going through t
src/model_loader/sklearn_bulk.cc:297
↓ 1 callers
Function
LoadText
src/model_loader/lightgbm.cc:223
↓ 1 callers
Function
LoadXGBoostModelFromJSONString
src/model_loader/xgboost_json.cc:80
↓ 1 callers
Function
LoadXGBoostModelFromUBJSONString
src/model_loader/xgboost_ubjson.cc:39
↓ 1 callers
Function
LoadXGBoostModelJSON
src/model_loader/xgboost_json.cc:41
↓ 1 callers
Function
LoadXGBoostModelUBJSON
src/model_loader/xgboost_ubjson.cc:32
↓ 1 callers
Method
Model
! \brief disable copy; use default move */
include/treelite/tree.h:457
↓ 1 callers
Function
NextNode
src/gtil/predict.cc:100
↓ 1 callers
Function
NextNodeCategorical
src/gtil/predict.cc:128
↓ 1 callers
Method
Null
src/model_loader/detail/xgboost_json/sax_adapters.cc:23
↓ 1 callers
Function
OmpGetThreadLimit
include/treelite/detail/threading_utils.h:48
↓ 1 callers
Function
OperatorFromString
! \brief Get Operator from string */
src/enum/operator.cc:34
↓ 1 callers
Function
OperatorToString
! \brief Get string representation of Operator */
src/enum/operator.cc:16
↓ 1 callers
Function
OutputLeafValue
src/gtil/predict.cc:219
↓ 1 callers
Function
OutputLeafVector
src/gtil/predict.cc:175
↓ 1 callers
Method
PeekRead
src/model_loader/xgboost_legacy.cc:97
↓ 1 callers
Function
PredictLeaf
src/gtil/predict.cc:326
↓ 1 callers
Function
PredictScoreByTree
src/gtil/predict.cc:348
↓ 1 callers
Function
ReadArrayFromStream
include/treelite/detail/serializer.h:169
↓ 1 callers
Method
RegisterCallBackLogInfo
include/treelite/logging.h:175
↓ 1 callers
Method
RegisterCallBackLogWarning
include/treelite/logging.h:181
↓ 1 callers
Method
Rethrow
! * \brief should be called from the main thread to rethrow the exception */
include/treelite/detail/omp_exception.h:51
↓ 1 callers
Function
SerializeModelParametersToJSON
src/json_serializer.cc:146
↓ 1 callers
Function
SerializeTaskParametersToJSON
src/json_serializer.cc:136
↓ 1 callers
Method
SerializeToBuffer
src/serializer.cc:482
↓ 1 callers
Method
SetCategoricalTest
include/treelite/detail/tree.h:138
↓ 1 callers
Method
SetDataCount
include/treelite/detail/tree.h:192
↓ 1 callers
Method
SetGain
include/treelite/detail/tree.h:202
↓ 1 callers
Method
SetHeaderField
src/field_accessor.cc:69
↓ 1 callers
Method
SetNumericalTest
include/treelite/detail/tree.h:127
↓ 1 callers
Method
SetSumHess
include/treelite/detail/tree.h:182
↓ 1 callers
Method
SetTreeField
src/field_accessor.cc:115
↓ 1 callers
Function
SetTreeFieldImpl
src/field_accessor.cc:191
↓ 1 callers
Method
SetValidationFlag
src/model_builder/model_builder.cc:89
↓ 1 callers
Function
SkipOptionalFieldInStream
include/treelite/detail/serializer.h:211
↓ 1 callers
Function
TaskTypeFromString
src/enum/task_type.cc:32
↓ 1 callers
Function
TaskTypeToString
src/enum/task_type.cc:15
↓ 1 callers
Function
TestRoundTrip
tests/cpp/test_model_concat.cc:20
↓ 1 callers
Function
TreeNodeTypeToString
src/enum/tree_node_type.cc:15
↓ 1 callers
Function
TreeliteAPISetLastError
src/c_api/c_api_error.cc:29
↓ 1 callers
Function
TreeliteDeleteModelBuilder
src/c_api/model_builder.cc:24
↓ 1 callers
Function
TreeliteGTILGetOutputShape
src/c_api/gtil.cc:32
↓ 1 callers
Function
TreeliteGTILParseConfig
src/c_api/gtil.cc:19
↓ 1 callers
Function
TreeliteGTILPredict
src/c_api/gtil.cc:44
↓ 1 callers
Function
TreeliteGetModelBuilder
src/c_api/model_builder.cc:17
↓ 1 callers
Function
TreeliteLoadXGBoostModelFromJSONString
src/c_api/model_loader.cc:63
↓ 1 callers
Function
TreeliteLoadXGBoostModelJSON
src/c_api/model_loader.cc:54
↓ 1 callers
Function
TreeliteModelBuilderCommitModel
src/c_api/model_builder.cc:126
↓ 1 callers
Function
TreeliteModelBuilderEndTree
src/c_api/model_builder.cc:37
↓ 1 callers
Function
TreeliteModelBuilderNumericalTest
src/c_api/model_builder.cc:58
↓ 1 callers
Function
TreeliteModelBuilderStartTree
src/c_api/model_builder.cc:30
↓ 1 callers
Function
WriteArrayToStream
include/treelite/detail/serializer.h:181
↓ 1 callers
Function
WriteScalarToStream
include/treelite/detail/serializer.h:163
↓ 1 callers
Function
WriteStringToStream
include/treelite/detail/serializer.h:201
↓ 1 callers
Function
_detect_xgboost_format
Inspect the first few bytes of an XGBoost model and heuristically determine whether it's using the JSON or UBJSON format. Parameters
python/treelite/frontend.py:348
↓ 1 callers
Function
_export_tree
( model, *, tree_id, n_features, n_classes, n_targets, tree_depths, subestimator_class )
python/treelite/sklearn/exporter.py:56
↓ 1 callers
Function
_import_hist_gradient_boosting
Load HistGradientBoostingClassifier / HistGradientBoostingRegressor
python/treelite/sklearn/importer.py:355
↓ 1 callers
Function
_require_keyword_args
Decorator for methods that issues warnings for positional arguments Using the keyword-only argument syntax in pep 3102, arguments after the *
python/treelite/util.py:81
↓ 1 callers
Method
_set_config_setting
(self, config_settings: Dict[str, Any])
python/packager/build_config.py:21
↓ 1 callers
Function
_train_model
(X, y)
tests/serializer/compatibility_tester.py:19
↓ 1 callers
Function
build_libtreelite
Build libtreelite in a temporary directory and obtain the path to built libtreelite
python/packager/nativelib.py:30
↓ 1 callers
Function
bytes_from_string_and_size
Copy `size` bytes from `ptr` to create a new python `bytes` object
python/treelite/util.py:43
↓ 1 callers
Function
calculate_depths
Fill in an array of isolation depths for a scikit-learn isolation forest model
python/treelite/sklearn/isolation_forest.py:20
↓ 1 callers
Function
cd
Temporarily change working directory
tests/ci_build/rename_whl.py:10
↓ 1 callers
Function
check_path
Ensure that this script is run from the root directory
dev/prepare_pypi_release.py:125
↓ 1 callers
Function
copy_cpp_src_tree
Copy C++ source tree into build directory
python/packager/sdist.py:11
↓ 1 callers
Method
deserialize_bytes
Deserialize (recover) the model from a byte sequence. It is expected that the byte sequence was generated by a call to the :py:func:`
python/treelite/model.py:299
← previous
next →
201–300 of 686, ranked by callers