MCPcopy Create free account

hub / github.com/dmlc/treelite / functions

Functions686 in github.com/dmlc/treelite

↓ 74 callersMethodPushBack
include/treelite/detail/contiguous_array.h:188
↓ 56 callersMethodStartNode
src/model_builder/model_builder.cc:155
↓ 55 callersMethodEndNode
src/model_builder/model_builder.cc:168
↓ 55 callersFunction_check_call
Check the return value of C API call This function will raise exception when error occurs. Wrap every API call with this function. Param
python/treelite/core.py:62
↓ 46 callersMethodas_c_array
Prepare the collection to pass as an argument of a C function
python/treelite/sklearn/importer.py:53
↓ 37 callersMethodClear
include/treelite/detail/contiguous_array.h:182
↓ 37 callersMethodLeafScalar
src/model_builder/model_builder.cc:214
↓ 34 callersMethodKey
src/model_loader/detail/xgboost_json/sax_adapters.cc:64
↓ 32 callersMethodDeserializeArray
include/treelite/detail/serializer_mixins.h:60
↓ 30 callersMethodDeserializeScalar
include/treelite/detail/serializer_mixins.h:51
↓ 30 callersMethodshould_ignore_upcoming_value
Perform this check at the top of every value handling function, to ignore extra fields from the JSON string
src/model_loader/detail/xgboost_json/delegated_handler.h:191
↓ 29 callersFunctionGetPyBufferFromArray
include/treelite/detail/serializer.h:30
↓ 29 callersMethodSize
include/treelite/detail/contiguous_array.h:138
↓ 29 callersMethodget_field
Get a field Parameters ---------- name: Name of the field. Consult :doc:`the model spec </serialization/
python/treelite/model.py:519
↓ 28 callersMethodStartTree
src/model_builder/model_builder.cc:95
↓ 27 callersMethodCommitModel
src/model_builder/model_builder.cc:279
↓ 27 callersMethodEndTree
src/model_builder/model_builder.cc:104
↓ 26 callersFunctionInitArrayFromPyBufferWithCopy
include/treelite/detail/serializer.h:119
↓ 26 callersMethodSerializeArray
include/treelite/detail/serializer_mixins.h:147
↓ 26 callersFunctionc_str
Convert a Python string to C string
python/treelite/util.py:38
↓ 24 callersMethodNumericalTest
src/model_builder/model_builder.cc:173
↓ 23 callersFunctionGetModelBuilder
src/model_builder/model_builder.cc:393
↓ 20 callersMethodReserve
include/treelite/detail/contiguous_array.h:148
↓ 20 callersMethodset_field
Set a field Parameters ---------- name: Name of the field. Consult :doc:`the model spec </serialization/
python/treelite/model.py:545
↓ 19 callersMethodstart_node
Start a new node Parameters ---------- node_key: Integer key that unique identifies the node
python/treelite/model_builder.py:197
↓ 18 callersFunctionGetPyBufferFromScalar
include/treelite/detail/serializer.h:75
↓ 17 callersMethodDumpAsJSON
src/json_serializer.cc:219
↓ 17 callersMethodSerializeScalar
include/treelite/detail/serializer_mixins.h:137
↓ 17 callersMethodend_node
End the current node
python/treelite/model_builder.py:210
↓ 14 callersMethodData
include/treelite/detail/contiguous_array.h:108
↓ 13 callersMethodGetNumTree
include/treelite/tree.h:478
↓ 13 callersMethodadd
Add an array to the collection
python/treelite/sklearn/importer.py:42
↓ 13 callersFunctionc_array
Convert a Python byte array to C array WARNING ------- DO NOT USE THIS FUNCTION if performance is critical. Instead, use np.array(*)
python/treelite/util.py:56
↓ 12 callersMethodRead
src/model_loader/xgboost_legacy.cc:67
↓ 12 callersFunctionTemporaryDirectory
Simulate the effect of 'ignore_cleanup_errors' parameter of tempfile.TemporaryDirectory. The parameter is only available for Python >= 3.10.
tests/python/util.py:124
↓ 12 callersFunctionTypeInfoToString
src/enum/typeinfo.cc:15
↓ 12 callersMethodleaf
Declare the current node as a leaf node Parameters ---------- leaf_value: Value of leaf output
python/treelite/model_builder.py:302
↓ 12 callersFunctionmake_tree_stump
()
tests/python/test_model_concat.py:22
↓ 10 callersMethodEmpty
include/treelite/detail/contiguous_array.h:143
↓ 10 callersMethodIsLeaf
src/serializer.cc:75
↓ 9 callersFunctionAssertDocumentValid
tests/cpp/test_model_builder.cc:29
↓ 9 callersFunctionInitScalarFromPyBuffer
include/treelite/detail/serializer.h:131
↓ 9 callersMethodLeafVector
! * \brief get leaf vector of the leaf node; useful for multi-class random forest classifier * \param nid ID of node being queried */
include/treelite/tree.h:218
↓ 9 callersFunctionto_categorical
Generate a Pandas DataFrame with a mix of numerical and categorical columns. Also returns a NumPy array whose value is equal to the DataFrame, exc
tests/python/util.py:43
↓ 8 callersMethodCategoricalTest
src/model_builder/model_builder.cc:192
↓ 8 callersMethodInitializeMetadata
src/model_builder/model_builder.cc:290
↓ 8 callersMethodLeafValue
src/serializer.cc:78
↓ 8 callersFunctionParallelFor
include/treelite/detail/threading_utils.h:111
↓ 8 callersFunctioncopy_with_logging
Call shutil.copy() with logging
python/packager/util.py:18
↓ 8 callersFunctionpy_str
Convert C string back to Python string
python/treelite/util.py:51
↓ 7 callersMethodDouble
src/model_loader/detail/xgboost_json/sax_adapters.cc:47
↓ 7 callersMethodGain
src/model_builder/model_builder.cc:258
↓ 7 callersMethodInt
src/model_loader/detail/xgboost_json/sax_adapters.cc:31
↓ 7 callersFunctionOpenFileForReadAsStream
include/treelite/detail/file_utils.h:20
↓ 7 callersFunctionPredict
src/gtil/predict.cc:399
↓ 7 callersMethodResize
include/treelite/detail/contiguous_array.h:157
↓ 7 callersFunctionWriteArray
src/json_serializer.cc:59
↓ 7 callersFunctionWriteString
src/json_serializer.cc:77
↓ 7 callersFunction_lib_name
Return platform dependent shared object name.
python/packager/nativelib.py:17
↓ 7 callersMethodstart_tree
Start a new tree
python/treelite/model_builder.py:189
↓ 6 callersMethodDataCount
src/model_builder/model_builder.cc:265
↓ 6 callersMethodEndArray
src/model_loader/detail/xgboost_json/sax_adapters.cc:76
↓ 6 callersMethodGetLeafOutputType
include/treelite/tree.h:475
↓ 6 callersMethodGetThresholdType
include/treelite/tree.h:472
↓ 6 callersMethodLeafVector
src/model_builder/model_builder.cc:226
↓ 6 callersMethodLeftChild
Getters **/ ! * \brief Index of the node's left child * \param nid ID of node being queried */
include/treelite/tree.h:169
↓ 6 callersMethodRightChild
! * \brief Index of the node's right child * \param nid ID of node being queried */
include/treelite/tree.h:176
↓ 6 callersMethodRun
include/treelite/detail/omp_exception.h:32
↓ 6 callersMethodStartArray
src/model_loader/detail/xgboost_json/sax_adapters.cc:72
↓ 6 callersFunctionStringStartsWith
src/model_loader/detail/string_utils.h:16
↓ 6 callersFunction_get_model_filename
(name, model_format)
tests/python/test_xgboost_integration.py:33
↓ 6 callersMethodend_tree
End the current tree
python/treelite/model_builder.py:193
↓ 6 callersMethodfrom_xgboost
Deprecated; please use :py:meth:`~treelite.frontend.from_xgboost` instead.
python/treelite/model.py:137
↓ 6 callersMethodnumerical_test
Declare the current node as a numerical test node, where the test is of form [feature value] [op] [threshold]. Data points for which
python/treelite/model_builder.py:215
↓ 6 callersFunctionomp_get_thread_num
include/treelite/detail/threading_utils.h:36
↓ 5 callersMethodAsVector
include/treelite/detail/contiguous_array.h:243
↓ 5 callersMethodBool
src/model_loader/detail/xgboost_json/sax_adapters.cc:27
↓ 5 callersMethodGain
! * \brief Get gain value * \param nid ID of node being queried */
include/treelite/tree.h:318
↓ 5 callersFunctionParseAttributes
src/model_builder/detail/json_parsing.h:212
↓ 5 callersMethodSumHess
src/model_builder/model_builder.cc:272
↓ 5 callersFunctionTestRoundTrip
tests/cpp/test_serializer.cc:33
↓ 5 callersFunction_ensure_scalar_int
(x: Any)
python/treelite/sklearn/exporter.py:12
↓ 5 callersFunction_get_limits
Try to find the strategy's limits. Raises AttributeError if limits cannot be determined. Credit: Carl Simon Adorf (@csadorf) https://gith
tests/python/hypothesis_util.py:15
↓ 5 callersFunction_qualify_path
(prefix, path)
tests/python/metadata.py:85
↓ 5 callersMethodasdict
Convert to dictionary
python/treelite/model_builder.py:43
↓ 5 callersMethodcommit
Conclude model building and obtain the final model object. Returns ------- model : :py:class:`Model` Fin
python/treelite/model_builder.py:387
↓ 5 callersMethoddump_as_json
Dump the model as a JSON string. This is useful for inspecting details of the tree ensemble model. Parameters ------
python/treelite/model.py:171
↓ 5 callersFunctioninvalid_numerical_test
(left_child_key, right_child_key)
tests/python/test_model_builder.py:66
↓ 5 callersFunctionraise_not_rf_error
(reason)
python/treelite/sklearn/exporter.py:180
↓ 5 callersMethodupdate
Parse config_settings from Pip (or other PEP 517 frontend)
python/packager/build_config.py:33
↓ 4 callersMethodClone
include/treelite/detail/tree.h:33
↓ 4 callersMethodEndObject
src/model_loader/detail/xgboost_json/sax_adapters.cc:68
↓ 4 callersMethodExtend
include/treelite/detail/contiguous_array.h:201
↓ 4 callersFunctionGetOutputShape
src/gtil/output_shape.cc:17
↓ 4 callersFunctionLoadSKLearnModel
src/model_loader/sklearn.cc:201
↓ 4 callersMethodSerializeHeader
src/serializer.cc:91
↓ 4 callersMethodSerializeToStream
src/serializer.cc:475
↓ 4 callersMethodSerializeTrees
src/serializer.cc:128
↓ 4 callersFunctionSplit
src/model_loader/lightgbm.cc:157
↓ 4 callersMethodSplitIndex
src/serializer.cc:72
next →1–100 of 686, ranked by callers