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
Function
TreeliteDeserializeModelFromPyBuffer
src/c_api/serializer.cc:77
Function
TreeliteDetectXGBoostFormat
src/c_api/model_loader.cc:90
Function
TreeliteDumpAsJSON
src/c_api/model.cc:18
Function
TreeliteFreeModel
src/c_api/model.cc:70
Function
TreeliteGTILDeleteConfig
src/c_api/gtil.cc:26
Function
TreeliteGTILPredictSparse
src/c_api/gtil.cc:62
Function
TreeliteGetHeaderField
src/c_api/field_accessor.cc:12
Function
TreeliteGetInputType
src/c_api/model.cc:27
Function
TreeliteGetLastError
src/c_api/c_api_error.cc:25
Function
TreeliteGetOutputType
src/c_api/model.cc:36
Function
TreeliteGetTreeDepth
src/c_api/model.cc:76
Function
TreeliteGetTreeField
src/c_api/field_accessor.cc:20
Function
TreeliteLoadLightGBMModel
src/c_api/model_loader.cc:98
Function
TreeliteLoadLightGBMModelFromString
src/c_api/model_loader.cc:107
Function
TreeliteLoadSKLearnGradientBoostingClassifier
src/c_api/sklearn.cc:70
Function
TreeliteLoadSKLearnGradientBoostingRegressor
src/c_api/sklearn.cc:56
Function
TreeliteLoadSKLearnHistGradientBoostingClassifier
src/c_api/sklearn.cc:99
Function
TreeliteLoadSKLearnHistGradientBoostingRegressor
src/c_api/sklearn.cc:84
Function
TreeliteLoadSKLearnIsolationForest
src/c_api/sklearn.cc:28
Function
TreeliteLoadSKLearnRandomForestClassifier
src/c_api/sklearn.cc:42
Function
TreeliteLoadSKLearnRandomForestRegressor
src/c_api/sklearn.cc:15
Function
TreeliteLoadXGBoostModel
src/c_api/model_loader.cc:40
Function
TreeliteLoadXGBoostModelFromString
src/c_api/model_loader.cc:47
Function
TreeliteLoadXGBoostModelFromUBJSONString
src/c_api/model_loader.cc:81
Function
TreeliteLoadXGBoostModelLegacyBinary
src/c_api/model_loader.cc:20
Function
TreeliteLoadXGBoostModelLegacyBinaryFromMemoryBuffer
src/c_api/model_loader.cc:30
Function
TreeliteLoadXGBoostModelUBJSON
src/c_api/model_loader.cc:72
Function
TreeliteModelBuilderCategoricalTest
src/c_api/model_builder.cc:69
Function
TreeliteModelBuilderDataCount
src/c_api/model_builder.cc:111
Function
TreeliteModelBuilderGain
src/c_api/model_builder.cc:104
Function
TreeliteModelBuilderLeafVectorFloat32
src/c_api/model_builder.cc:88
Function
TreeliteModelBuilderLeafVectorFloat64
src/c_api/model_builder.cc:96
Function
TreeliteModelBuilderSumHess
src/c_api/model_builder.cc:119
Function
TreeliteQueryNumFeature
src/c_api/model.cc:52
Function
TreeliteQueryNumTree
src/c_api/model.cc:45
Function
TreeliteQueryTreeliteVersion
src/c_api/c_api_error.cc:33
Function
TreeliteRegisterLogCallback
src/c_api/logging.cc:12
Function
TreeliteRegisterWarningCallback
src/c_api/logging.cc:19
Function
TreeliteSerializeModelToBytes
src/c_api/serializer.cc:36
Function
TreeliteSerializeModelToFile
src/c_api/serializer.cc:20
Function
TreeliteSerializeModelToPyBuffer
src/c_api/serializer.cc:60
Function
TreeliteSetHeaderField
src/c_api/field_accessor.cc:28
Function
TreeliteSetTreeField
src/c_api/field_accessor.cc:36
Function
TypeInfoFromType
include/treelite/enum/typeinfo.h:35
Method
Uint64
src/model_loader/detail/xgboost_json/delegated_handler.cc:86
Method
Uint64
src/model_loader/detail/xgboost_json/delegated_handler.h:329
Function
WriteNode
src/json_serializer.cc:82
Method
__del__
(self)
python/treelite/model_builder.py:179
Method
__del__
(self)
python/treelite/model.py:29
Method
__del__
(self)
python/treelite/gtil/gtil.py:44
Method
__init__
( self, *, threshold_type: str, leaf_output_type: str, metadata: Metad
python/treelite/model_builder.py:144
Method
__init__
(self, *, handle: Optional[Any] = None)
python/treelite/model.py:26
Method
__init__
(self, model: Model)
python/treelite/model.py:443
Method
__init__
(self, model: Model, *, tree_id: int)
python/treelite/model.py:515
Method
__init__
(self, *, dtype)
python/treelite/sklearn/importer.py:20
Method
__post_init__
(self)
python/treelite/gtil/gtil.py:36
Function
_log_callback
Redirect logs from native library into Python console
python/treelite/core.py:17
Function
_warn_callback
Redirect warnings from native library into Python console
python/treelite/core.py:23
Method
asdict
Convert to dictionary
python/treelite/model_builder.py:75
Method
asdict
Convert to dictionary
python/treelite/model_builder.py:104
Method
assign_value
src/model_loader/detail/xgboost_json/delegated_handler.cc:176
Method
binary
src/model_loader/detail/xgboost_json/sax_adapters.cc:108
Method
boolean
src/model_loader/detail/xgboost_json/sax_adapters.cc:88
Function
build_editable
Build an editable installation. We mostly delegate to Hatchling.
python/packager/pep517.py:153
Function
build_sdist
Build a source distribution
python/packager/pep517.py:109
Function
build_wheel
Build a wheel
python/packager/pep517.py:50
Method
categorical_test
Declare the current node as a categorical test node, where the test is of form [feature value] \\in [category list]. Paramet
python/treelite/model_builder.py:259
Method
cdefault
src/model_loader/xgboost_legacy.cc:211
Method
check_cur_key
src/model_loader/detail/xgboost_json/delegated_handler.cc:171
Method
create
src/model_loader/detail/xgboost_json/delegated_handler.cc:215
Method
create_empty
src/model_loader/detail/xgboost_json/delegated_handler.cc:204
Method
data_count
Specify the number of data points (samples) that are mapped to the current node. Parameters ---------- data_count:
python/treelite/model_builder.py:354
Method
deprecation_error
(alt: str)
python/treelite/model.py:122
Function
does_not_raise
Placeholder to indicate that a section of code is not expected to raise any exception
tests/python/util.py:37
Method
end_array
src/model_loader/detail/xgboost_json/sax_adapters.cc:129
Method
end_object
src/model_loader/detail/xgboost_json/sax_adapters.cc:121
Function
exponential
src/gtil/postprocessor.cc:40
Function
exponential_standard_ratio
src/gtil/postprocessor.cc:45
Function
export_model
Export a model as a scikit-learn RandomForest. Note ---- Currently only random forests can be exported as scikit-learn model objects
python/treelite/sklearn/exporter.py:128
Function
from_lightgbm
Load a tree ensemble model from a LightGBM Booster object Parameters ---------- booster : object of type :py:class:`lightgbm.Booster
python/treelite/frontend.py:314
Function
from_xgboost
Load a tree ensemble model from an XGBoost Booster object Parameters ---------- booster : Object of type :py:class:`xgboost.Booster`
python/treelite/frontend.py:178
Method
gain
Specify the gain (loss reduction) that's resulted from the current split. Parameters ---------- gain: Ga
python/treelite/model_builder.py:338
Method
get_field
Get a field Parameters ---------- name: Name of the field. Consult :doc:`the model spec </serialization/
python/treelite/model.py:446
Method
handle
Access the handle to the associated C++ object
python/treelite/model_builder.py:185
Method
handle
Access the handle to the associated C++ object
python/treelite/model.py:35
Method
handle
Access the handle to the associated C++ object
python/treelite/gtil/gtil.py:49
Function
hinge
src/gtil/postprocessor.cc:29
Function
identity
src/gtil/postprocessor.cc:20
Function
identity_multiclass
src/gtil/postprocessor.cc:55
Function
import_model
Load a tree ensemble model from a scikit-learn model object Note ---- For :py:class:`~sklearn.ensemble.IsolationForest`, the loaded
python/treelite/sklearn/importer.py:62
Function
import_model_with_model_builder
This function was removed in Treelite 4.0; please use :py:meth:`~treelite.sklearn.import_model` instead.
python/treelite/sklearn/__init__.py:7
Method
initialize
This step ccurs immediately before each build.
python/hatch_build.py:21
Function
inner_f
(*args: Any, **kwargs: Any)
python/treelite/util.py:117
Method
input_type
Input type
python/treelite/model.py:58
Method
is_recognized_key
src/model_loader/detail/xgboost_json/delegated_handler.cc:331
Method
is_recognized_key
\brief Check if a given key is recognized by the handler * \param key Key to look up */
src/model_loader/detail/xgboost_json/delegated_handler.h:185
Method
is_root
src/model_loader/xgboost_legacy.cc:232
Method
key
src/model_loader/detail/xgboost_json/sax_adapters.cc:133
Function
load_lightgbm_model
Load a tree ensemble model from a LightGBM model file. Parameters ---------- filename : Path to model file Returns
python/treelite/frontend.py:147
Function
load_xgboost_model
Load a tree ensemble model from XGBoost model, stored using JSON or UBJSON format. Parameters ---------- filename : Path to
python/treelite/frontend.py:60
← previous
next →
501–600 of 686, ranked by callers