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
↓ 4 callers
Method
StartObject
src/model_loader/detail/xgboost_json/sax_adapters.cc:60
↓ 4 callers
Method
Threshold
src/serializer.cc:81
↓ 4 callers
Function
TypeInfoFromString
src/enum/typeinfo.cc:31
↓ 4 callers
Method
Uint64
src/model_loader/detail/xgboost_json/sax_adapters.cc:43
↓ 4 callers
Function
_ensure_numpy
(x: Any)
python/treelite/sklearn/exporter.py:22
↓ 4 callers
Function
copytree_with_logging
Call shutil.copytree() with logging
python/packager/util.py:10
↓ 4 callers
Method
get_tree_accessor
Obtain accessor for fields in a tree. See :ref:`field_accessors` for more details. Parameters ---------- tre
python/treelite/model.py:218
↓ 4 callers
Function
standard_regression_datasets
Returns a strategy to generate regression problem input datasets. Note: This function uses the sklearn.datasets.make_regression function
tests/python/hypothesis_util.py:188
↓ 4 callers
Function
typestr_to_ctypes_type
Obtain ctypes type corresponding to a given Type str
python/treelite/util.py:28
↓ 3 callers
Function
AssertJSONStringsEqual
tests/cpp/test_model_builder.cc:35
↓ 3 callers
Function
ConcatenateModelObjects
src/model_concat.cc:19
↓ 3 callers
Function
EvaluateTree
src/gtil/predict.cc:153
↓ 3 callers
Function
GetPyBufferFromString
include/treelite/detail/serializer.h:79
↓ 3 callers
Method
GetRow
src/gtil/predict.cc:48
↓ 3 callers
Method
HasLeafVector
! * \brief Tests whether the leaf node has a non-empty leaf vector * \param nid ID of node being queried */
include/treelite/tree.h:233
↓ 3 callers
Function
InitStringFromPyBuffer
include/treelite/detail/serializer.h:126
↓ 3 callers
Method
Int64
src/model_loader/detail/xgboost_json/sax_adapters.cc:39
↓ 3 callers
Function
LoadHistGradientBoosting
src/model_loader/sklearn.cc:353
↓ 3 callers
Function
MaxNumThread
include/treelite/detail/threading_utils.h:59
↓ 3 callers
Method
NodeType
! * \brief Get the type of a node * \param nid ID of node being queried */
include/treelite/tree.h:275
↓ 3 callers
Function
OpenFileForWriteAsStream
include/treelite/detail/file_utils.h:40
↓ 3 callers
Function
ParseBaseScore
src/model_loader/detail/xgboost.cc:62
↓ 3 callers
Function
ParseMetadata
src/model_builder/detail/json_parsing.h:168
↓ 3 callers
Function
ParsePostProcessorFunc
src/model_builder/detail/json_parsing.h:189
↓ 3 callers
Function
ParseTreeAnnotation
src/model_builder/detail/json_parsing.h:181
↓ 3 callers
Function
ReadScalarFromStream
include/treelite/detail/serializer.h:157
↓ 3 callers
Method
SerializeToPyBuffer
src/serializer.cc:458
↓ 3 callers
Method
SetChildren
Setters **/ ! * \brief Identify two child nodes of the node * \param nid ID of node being modified * \param left_child ID of the left child n
include/treelite/tree.h:344
↓ 3 callers
Method
SetLeaf
include/treelite/detail/tree.h:158
↓ 3 callers
Method
SkipOptionalField
include/treelite/detail/serializer_mixins.h:64
↓ 3 callers
Method
String
src/model_loader/detail/xgboost_json/sax_adapters.cc:56
↓ 3 callers
Function
TreeliteModelBuilderEndNode
src/c_api/model_builder.cc:51
↓ 3 callers
Function
TreeliteModelBuilderStartNode
src/c_api/model_builder.cc:44
↓ 3 callers
Method
Uint
src/model_loader/detail/xgboost_json/sax_adapters.cc:35
↓ 3 callers
Function
WriteElement
src/json_serializer.cc:30
↓ 3 callers
Function
_build
(*, generator: str)
python/packager/nativelib.py:44
↓ 3 callers
Function
_normalize_path
Fully expand a path and convert it to an absolute path
python/treelite/frontend.py:17
↓ 3 callers
Function
_numpy2pybuffer
(array: np.ndarray)
python/treelite/model.py:409
↓ 3 callers
Function
_predict_impl
( model: Model, data: Union[np.ndarray, csr_matrix], *, config: GTILConfig )
python/treelite/gtil/gtil.py:150
↓ 3 callers
Function
_pybuffer2numpy
(frame: _TreelitePyBufferFrame)
python/treelite/model.py:371
↓ 3 callers
Method
concatenate
Concatenate multiple model objects into a single model object by copying all member trees into the destination model object.
python/treelite/model.py:76
↓ 3 callers
Function
get_extra_field_value
()
tests/python/test_xgboost_integration.py:367
↓ 3 callers
Method
get_header_accessor
Obtain accessor for fields in the header. See :ref:`field_accessors` for more details.
python/treelite/model.py:211
↓ 3 callers
Function
is_readthedocs_build
()
docs/conf.py:44
↓ 3 callers
Function
parse_as_json
()
python/treelite/frontend.py:104
↓ 3 callers
Function
parse_as_ubjson
()
python/treelite/frontend.py:113
↓ 3 callers
Method
serialize
Serialize (persist) the model to a checkpoint file in the disk, using a fast binary representation. To recover the model from the che
python/treelite/model.py:230
↓ 3 callers
Function
typestr_to_numpy_type
Obtain ctypes type corresponding to a given Type str
python/treelite/util.py:33
↓ 2 callers
Function
CONSUME_BYTES
src/model_loader/xgboost_legacy.cc:145
↓ 2 callers
Method
CategoryList
! * \brief Get list of all categories belonging to the left/right child node. * See the category_list_right_child_ field of each test node to de
include/treelite/tree.h:259
↓ 2 callers
Method
CategoryListRightChild
! * \brief Test whether the list given by CategoryList(nid) is associated with the right * child node or the left child node * \param
include/treelite/tree.h:326
↓ 2 callers
Method
ComparisonOp
! * \brief Get comparison operator * \param nid ID of node being queried */
include/treelite/tree.h:247
↓ 2 callers
Method
DefaultLeft
src/serializer.cc:69
↓ 2 callers
Method
DeserializeHeaderAndCreateModel
src/serializer.cc:186
↓ 2 callers
Method
DeserializeString
include/treelite/detail/serializer_mixins.h:55
↓ 2 callers
Method
DeserializeTrees
src/serializer.cc:247
↓ 2 callers
Function
DumpModelAsJSON
src/json_serializer.cc:182
↓ 2 callers
Function
GetDecisionType
src/model_loader/lightgbm.cc:202
↓ 2 callers
Function
GetPostProcessor
Get correct postprocessor for prediction, depending on objective function
src/model_loader/detail/xgboost.cc:28
↓ 2 callers
Method
HandleMetadata
src/model_loader/sklearn.cc:37
↓ 2 callers
Function
LoadHistGradientBoostingImpl
src/model_loader/sklearn.cc:285
↓ 2 callers
Function
LoadXGBoostModelLegacyBinary
src/model_loader/xgboost_legacy.cc:41
↓ 2 callers
Function
OpenFileForReadAsFilePtr
include/treelite/detail/file_utils.h:59
↓ 2 callers
Function
ParseStream
src/model_loader/xgboost_json.cc:113
↓ 2 callers
Function
ParseStream
src/model_loader/xgboost_ubjson.cc:50
↓ 2 callers
Function
ParseStream
src/model_loader/lightgbm.cc:234
↓ 2 callers
Function
ParseStream
src/model_loader/xgboost_legacy.cc:318
↓ 2 callers
Function
PredictImpl
src/gtil/predict.cc:381
↓ 2 callers
Function
PredictRaw
src/gtil/predict.cc:232
↓ 2 callers
Function
PredictSparse
src/gtil/predict.cc:407
↓ 2 callers
Function
ReadStringFromStream
include/treelite/detail/serializer.h:191
↓ 2 callers
Method
SerializeString
include/treelite/detail/serializer_mixins.h:141
↓ 2 callers
Method
SetLeafVector
include/treelite/detail/tree.h:166
↓ 2 callers
Function
StringTrimFromEnd
src/model_loader/detail/string_utils.h:20
↓ 2 callers
Function
TransformBaseScoreToMargin
src/model_loader/detail/xgboost.cc:52
↓ 2 callers
Function
TreeliteModelBuilderLeafScalar
src/c_api/model_builder.cc:81
↓ 2 callers
Method
UseForeignBuffer
include/treelite/detail/contiguous_array.h:97
↓ 2 callers
Function
_fetch_data
()
tests/serializer/compatibility_tester.py:14
↓ 2 callers
Function
_load_lib
Load Treelite Library.
python/treelite/core.py:28
↓ 2 callers
Function
cd
Temporarily change working directory. TODO(hcho3): Remove this once we adopt Python 3.11, which implements contextlib.chdir.
python/packager/pep517.py:25
↓ 2 callers
Method
cleft
src/model_loader/xgboost_legacy.cc:205
↓ 2 callers
Method
cright
src/model_loader/xgboost_legacy.cc:208
↓ 2 callers
Method
default_left
src/model_loader/xgboost_legacy.cc:217
↓ 2 callers
Method
deserialize
Deserialize (recover) the model from a checkpoint file in the disk. It is expected that the file was generated by a call to the :py:f
python/treelite/model.py:270
↓ 2 callers
Function
download_wheels
Download all binary wheels. url_prefix is the URL for remote directory storing the release wheels
dev/prepare_pypi_release.py:53
↓ 2 callers
Function
expected_depth
Calculates the expected isolation depth for a remainder of uniform points
python/treelite/sklearn/isolation_forest.py:11
↓ 2 callers
Method
from_lightgbm
Deprecated; please use :py:meth:`~treelite.frontend.from_lightgbm` instead.
python/treelite/model.py:162
↓ 2 callers
Function
from_xgboost_json
Load a tree ensemble model from a string containing XGBoost JSON Parameters ---------- model_json_str : A string specifying
python/treelite/frontend.py:228
↓ 2 callers
Function
generate_data_for_squared_log_error
Generate data containing outliers.
tests/python/test_xgboost_integration.py:39
↓ 2 callers
Method
get_result
src/model_loader/detail/xgboost_json/delegated_handler.cc:229
↓ 2 callers
Function
load_txt
Get 1D array from text file
tests/python/util.py:15
↓ 2 callers
Function
locate_local_libtreelite
Locate libtreelite from the local project directory's lib/ subdirectory.
python/packager/nativelib.py:105
↓ 2 callers
Function
make_tree_stump
(left_child_val, right_child_val)
tests/python/test_field_accessor.py:33
↓ 2 callers
Function
make_tree_stump
(left_child_val, right_child_val)
tests/python/test_model_builder.py:107
↓ 2 callers
Method
push_delegate
src/model_loader/detail/xgboost_json/delegated_handler.cc:221
↓ 2 callers
Method
serialize_bytes
Serialize (persist) the model to a byte sequence, using a fast binary representation. To recover the model from the byte sequence, us
python/treelite/model.py:250
↓ 2 callers
Function
softmax
src/gtil/postprocessor.cc:58
↓ 2 callers
Function
treelite_serialize
Deserialize model from PyBuffer frames
tests/serializer/test_serializer.py:35
↓ 1 callers
Method
AllocNode
include/treelite/detail/tree.h:70
↓ 1 callers
Function
ApplyPostProcessor
src/gtil/predict.cc:308
← previous
next →
101–200 of 686, ranked by callers