MCPcopy Create free account

hub / github.com/dmlc/treelite / functions

Functions686 in github.com/dmlc/treelite

↓ 4 callersMethodStartObject
src/model_loader/detail/xgboost_json/sax_adapters.cc:60
↓ 4 callersMethodThreshold
src/serializer.cc:81
↓ 4 callersFunctionTypeInfoFromString
src/enum/typeinfo.cc:31
↓ 4 callersMethodUint64
src/model_loader/detail/xgboost_json/sax_adapters.cc:43
↓ 4 callersFunction_ensure_numpy
(x: Any)
python/treelite/sklearn/exporter.py:22
↓ 4 callersFunctioncopytree_with_logging
Call shutil.copytree() with logging
python/packager/util.py:10
↓ 4 callersMethodget_tree_accessor
Obtain accessor for fields in a tree. See :ref:`field_accessors` for more details. Parameters ---------- tre
python/treelite/model.py:218
↓ 4 callersFunctionstandard_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 callersFunctiontypestr_to_ctypes_type
Obtain ctypes type corresponding to a given Type str
python/treelite/util.py:28
↓ 3 callersFunctionAssertJSONStringsEqual
tests/cpp/test_model_builder.cc:35
↓ 3 callersFunctionConcatenateModelObjects
src/model_concat.cc:19
↓ 3 callersFunctionEvaluateTree
src/gtil/predict.cc:153
↓ 3 callersFunctionGetPyBufferFromString
include/treelite/detail/serializer.h:79
↓ 3 callersMethodGetRow
src/gtil/predict.cc:48
↓ 3 callersMethodHasLeafVector
! * \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 callersFunctionInitStringFromPyBuffer
include/treelite/detail/serializer.h:126
↓ 3 callersMethodInt64
src/model_loader/detail/xgboost_json/sax_adapters.cc:39
↓ 3 callersFunctionLoadHistGradientBoosting
src/model_loader/sklearn.cc:353
↓ 3 callersFunctionMaxNumThread
include/treelite/detail/threading_utils.h:59
↓ 3 callersMethodNodeType
! * \brief Get the type of a node * \param nid ID of node being queried */
include/treelite/tree.h:275
↓ 3 callersFunctionOpenFileForWriteAsStream
include/treelite/detail/file_utils.h:40
↓ 3 callersFunctionParseBaseScore
src/model_loader/detail/xgboost.cc:62
↓ 3 callersFunctionParseMetadata
src/model_builder/detail/json_parsing.h:168
↓ 3 callersFunctionParsePostProcessorFunc
src/model_builder/detail/json_parsing.h:189
↓ 3 callersFunctionParseTreeAnnotation
src/model_builder/detail/json_parsing.h:181
↓ 3 callersFunctionReadScalarFromStream
include/treelite/detail/serializer.h:157
↓ 3 callersMethodSerializeToPyBuffer
src/serializer.cc:458
↓ 3 callersMethodSetChildren
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 callersMethodSetLeaf
include/treelite/detail/tree.h:158
↓ 3 callersMethodSkipOptionalField
include/treelite/detail/serializer_mixins.h:64
↓ 3 callersMethodString
src/model_loader/detail/xgboost_json/sax_adapters.cc:56
↓ 3 callersFunctionTreeliteModelBuilderEndNode
src/c_api/model_builder.cc:51
↓ 3 callersFunctionTreeliteModelBuilderStartNode
src/c_api/model_builder.cc:44
↓ 3 callersMethodUint
src/model_loader/detail/xgboost_json/sax_adapters.cc:35
↓ 3 callersFunctionWriteElement
src/json_serializer.cc:30
↓ 3 callersFunction_build
(*, generator: str)
python/packager/nativelib.py:44
↓ 3 callersFunction_normalize_path
Fully expand a path and convert it to an absolute path
python/treelite/frontend.py:17
↓ 3 callersFunction_numpy2pybuffer
(array: np.ndarray)
python/treelite/model.py:409
↓ 3 callersFunction_predict_impl
( model: Model, data: Union[np.ndarray, csr_matrix], *, config: GTILConfig )
python/treelite/gtil/gtil.py:150
↓ 3 callersFunction_pybuffer2numpy
(frame: _TreelitePyBufferFrame)
python/treelite/model.py:371
↓ 3 callersMethodconcatenate
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 callersFunctionget_extra_field_value
()
tests/python/test_xgboost_integration.py:367
↓ 3 callersMethodget_header_accessor
Obtain accessor for fields in the header. See :ref:`field_accessors` for more details.
python/treelite/model.py:211
↓ 3 callersFunctionis_readthedocs_build
()
docs/conf.py:44
↓ 3 callersFunctionparse_as_json
()
python/treelite/frontend.py:104
↓ 3 callersFunctionparse_as_ubjson
()
python/treelite/frontend.py:113
↓ 3 callersMethodserialize
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 callersFunctiontypestr_to_numpy_type
Obtain ctypes type corresponding to a given Type str
python/treelite/util.py:33
↓ 2 callersFunctionCONSUME_BYTES
src/model_loader/xgboost_legacy.cc:145
↓ 2 callersMethodCategoryList
! * \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 callersMethodCategoryListRightChild
! * \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 callersMethodComparisonOp
! * \brief Get comparison operator * \param nid ID of node being queried */
include/treelite/tree.h:247
↓ 2 callersMethodDefaultLeft
src/serializer.cc:69
↓ 2 callersMethodDeserializeHeaderAndCreateModel
src/serializer.cc:186
↓ 2 callersMethodDeserializeString
include/treelite/detail/serializer_mixins.h:55
↓ 2 callersMethodDeserializeTrees
src/serializer.cc:247
↓ 2 callersFunctionDumpModelAsJSON
src/json_serializer.cc:182
↓ 2 callersFunctionGetDecisionType
src/model_loader/lightgbm.cc:202
↓ 2 callersFunctionGetPostProcessor
Get correct postprocessor for prediction, depending on objective function
src/model_loader/detail/xgboost.cc:28
↓ 2 callersMethodHandleMetadata
src/model_loader/sklearn.cc:37
↓ 2 callersFunctionLoadHistGradientBoostingImpl
src/model_loader/sklearn.cc:285
↓ 2 callersFunctionLoadXGBoostModelLegacyBinary
src/model_loader/xgboost_legacy.cc:41
↓ 2 callersFunctionOpenFileForReadAsFilePtr
include/treelite/detail/file_utils.h:59
↓ 2 callersFunctionParseStream
src/model_loader/xgboost_json.cc:113
↓ 2 callersFunctionParseStream
src/model_loader/xgboost_ubjson.cc:50
↓ 2 callersFunctionParseStream
src/model_loader/lightgbm.cc:234
↓ 2 callersFunctionParseStream
src/model_loader/xgboost_legacy.cc:318
↓ 2 callersFunctionPredictImpl
src/gtil/predict.cc:381
↓ 2 callersFunctionPredictRaw
src/gtil/predict.cc:232
↓ 2 callersFunctionPredictSparse
src/gtil/predict.cc:407
↓ 2 callersFunctionReadStringFromStream
include/treelite/detail/serializer.h:191
↓ 2 callersMethodSerializeString
include/treelite/detail/serializer_mixins.h:141
↓ 2 callersMethodSetLeafVector
include/treelite/detail/tree.h:166
↓ 2 callersFunctionStringTrimFromEnd
src/model_loader/detail/string_utils.h:20
↓ 2 callersFunctionTransformBaseScoreToMargin
src/model_loader/detail/xgboost.cc:52
↓ 2 callersFunctionTreeliteModelBuilderLeafScalar
src/c_api/model_builder.cc:81
↓ 2 callersMethodUseForeignBuffer
include/treelite/detail/contiguous_array.h:97
↓ 2 callersFunction_fetch_data
()
tests/serializer/compatibility_tester.py:14
↓ 2 callersFunction_load_lib
Load Treelite Library.
python/treelite/core.py:28
↓ 2 callersFunctioncd
Temporarily change working directory. TODO(hcho3): Remove this once we adopt Python 3.11, which implements contextlib.chdir.
python/packager/pep517.py:25
↓ 2 callersMethodcleft
src/model_loader/xgboost_legacy.cc:205
↓ 2 callersMethodcright
src/model_loader/xgboost_legacy.cc:208
↓ 2 callersMethoddefault_left
src/model_loader/xgboost_legacy.cc:217
↓ 2 callersMethoddeserialize
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 callersFunctiondownload_wheels
Download all binary wheels. url_prefix is the URL for remote directory storing the release wheels
dev/prepare_pypi_release.py:53
↓ 2 callersFunctionexpected_depth
Calculates the expected isolation depth for a remainder of uniform points
python/treelite/sklearn/isolation_forest.py:11
↓ 2 callersMethodfrom_lightgbm
Deprecated; please use :py:meth:`~treelite.frontend.from_lightgbm` instead.
python/treelite/model.py:162
↓ 2 callersFunctionfrom_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 callersFunctiongenerate_data_for_squared_log_error
Generate data containing outliers.
tests/python/test_xgboost_integration.py:39
↓ 2 callersMethodget_result
src/model_loader/detail/xgboost_json/delegated_handler.cc:229
↓ 2 callersFunctionload_txt
Get 1D array from text file
tests/python/util.py:15
↓ 2 callersFunctionlocate_local_libtreelite
Locate libtreelite from the local project directory's lib/ subdirectory.
python/packager/nativelib.py:105
↓ 2 callersFunctionmake_tree_stump
(left_child_val, right_child_val)
tests/python/test_field_accessor.py:33
↓ 2 callersFunctionmake_tree_stump
(left_child_val, right_child_val)
tests/python/test_model_builder.py:107
↓ 2 callersMethodpush_delegate
src/model_loader/detail/xgboost_json/delegated_handler.cc:221
↓ 2 callersMethodserialize_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 callersFunctionsoftmax
src/gtil/postprocessor.cc:58
↓ 2 callersFunctiontreelite_serialize
Deserialize model from PyBuffer frames
tests/serializer/test_serializer.py:35
↓ 1 callersMethodAllocNode
include/treelite/detail/tree.h:70
↓ 1 callersFunctionApplyPostProcessor
src/gtil/predict.cc:308
← previousnext →101–200 of 686, ranked by callers