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
load_xgboost_model_legacy_binary
Load a tree ensemble model from XGBoost model, stored using the legacy binary format. Note: new XGBoost models should be stored in the JS
python/treelite/frontend.py:25
Function
logarithm_one_plus_exp
src/gtil/postprocessor.cc:50
Function
main
tests/cpp/test_main.cc:9
Function
main
tests/example_app/example.cc:17
Function
main
tests/example_app/example.c:68
Method
make_shared_enabler
src/model_loader/detail/xgboost_json/delegated_handler.cc:207
Function
multiclass_ova
src/gtil/postprocessor.cc:78
Method
null
src/model_loader/detail/xgboost_json/sax_adapters.cc:84
Method
num_feature
Number of features used in the model
python/treelite/model.py:49
Method
num_tree
Number of decision trees in the model
python/treelite/model.py:40
Method
number_float
src/model_loader/detail/xgboost_json/sax_adapters.cc:100
Method
number_integer
src/model_loader/detail/xgboost_json/sax_adapters.cc:92
Method
number_unsigned
src/model_loader/detail/xgboost_json/sax_adapters.cc:96
Method
operator=
include/treelite/logging.h:121
Method
operator==
include/treelite/detail/contiguous_array.h:251
Method
output_type
Output type
python/treelite/model.py:67
Method
parent
src/model_loader/xgboost_legacy.cc:229
Method
parse_error
src/model_loader/detail/xgboost_json/sax_adapters.cc:137
Method
pop_handler
src/model_loader/detail/xgboost_json/delegated_handler.cc:143
Function
predict
Predict with a Treelite model using the General Tree Inference Library (GTIL). Parameters ---------- model : :py:class:`Model` objec
python/treelite/gtil/gtil.py:54
Function
predict_leaf
Predict with a Treelite model, outputting the leaf node's ID for each row. Parameters ---------- model : :py:class:`Model` object
python/treelite/gtil/gtil.py:87
Function
predict_per_tree
Predict with a Treelite model and output prediction of each tree. This function computes one or more margin scores per tree. Parameters
python/treelite/gtil/gtil.py:123
Method
push_handler
src/model_loader/detail/xgboost_json/delegated_handler.h:135
Method
push_key_handler
src/model_loader/detail/xgboost_json/delegated_handler.h:150
Method
set_cur_key
src/model_loader/detail/xgboost_json/delegated_handler.cc:152
Method
set_field
Set a field Parameters ---------- name: Name of the field. Consult :doc:`the model spec </serialization/
python/treelite/model.py:477
Function
setup
(app)
docs/conf.py:228
Function
show_progress
Show file download progress.
dev/prepare_pypi_release.py:24
Function
sigmoid
src/gtil/postprocessor.cc:34
Function
signed_square
src/gtil/postprocessor.cc:23
Function
standard_multi_target_binary_classification_datasets
Returns a strategy to generate datasets for multi-target binary classification, where each target is associated with a binary class label.
tests/python/hypothesis_util.py:268
Function
standard_settings
Default hypothesis settings
tests/python/hypothesis_util.py:324
Method
start_array
src/model_loader/detail/xgboost_json/sax_adapters.cc:125
Method
start_object
src/model_loader/detail/xgboost_json/sax_adapters.cc:117
Method
store_int
src/model_loader/detail/xgboost_json/delegated_handler.h:312
Method
store_string
src/model_loader/detail/xgboost_json/delegated_handler.h:349
Method
sum_hess
Specify the weighted sample count or the sum of Hessians for the data points that are mapped to the current node. Parameters
python/treelite/model_builder.py:370
Function
test_categorical_encoding
Treelite should throw an exception when XGBoost model uses categorical encoding
tests/python/test_xgboost_integration.py:611
Function
test_extra_field_in_xgb_json
Test if we can handle extra fields in XGBoost JSON model file Insert an extra field at a random place and then load the model into Treelite,
tests/python/test_xgboost_integration.py:334
Function
test_getter_setter
Test getter and setter methods
tests/python/test_field_accessor.py:15
Function
test_invalid_node_id
Test for invalid node IDs
tests/python/test_model_builder.py:44
Function
test_leaf_vector_rf
Test a small random forest with leaf vector output
tests/python/test_model_builder.py:89
Function
test_lightgbm_binary_classification
Test LightGBM binary classifier
tests/python/test_lightgbm_integration.py:101
Function
test_lightgbm_categorical_data
Test LightGBM with toy categorical data
tests/python/test_lightgbm_integration.py:209
Function
test_lightgbm_deep_tree
Test LightGBM model with depth 32+
tests/python/test_lightgbm_integration.py:272
Function
test_lightgbm_multiclass_classification
Test LightGBM multi-class classifier
tests/python/test_lightgbm_integration.py:158
Function
test_lightgbm_regression
Test LightGBM regressor
tests/python/test_lightgbm_integration.py:46
Function
test_lightgbm_sparse_categorical_model
Test LightGBM with high-cardinality categorical features
tests/python/test_lightgbm_integration.py:252
Function
test_lightgbm_sparse_ranking_model
Generate a LightGBM ranking model with highly sparse data.
tests/python/test_lightgbm_integration.py:221
Function
test_load_old_xgboost_model
Ensure that Treelite can load old XGBoost models
tests/python/test_xgboost_integration.py:598
Function
test_model_concat_with_mismatched_tree_types
Test model concatenation with a tree stump
tests/python/test_model_concat.py:91
Function
test_model_concat_with_tree_stump
Test model concatenation with a tree stump
tests/python/test_model_concat.py:18
Function
test_orphaned_nodes
Test for orphaned nodes
tests/python/test_model_builder.py:16
Function
test_predict_special_with_binary_classifier
Test predict_leaf / predict_per_tree with XGBoost binary classifier
tests/python/test_gtil.py:77
Function
test_predict_special_with_multiclass_classifier_grove_per_class
Test predict_leaf / predict_per_tree with XGBoost multiclass classifier (grove-per-class)
tests/python/test_gtil.py:120
Function
test_predict_special_with_regressor
Test predict_leaf / predict_per_tree with XGBoost regressor
tests/python/test_gtil.py:37
Function
test_serialize_as_buffer
Test whether Treelite objects can be serialized to a buffer
tests/serializer/test_serializer.py:54
Function
test_serialize_as_bytes
Test whether Treelite objects can be serialized to a byte sequence
tests/python/test_serializer.py:58
Function
test_serialize_as_checkpoint
Test whether Treelite objects can be serialized to a checkpoint
tests/python/test_serializer.py:112
Function
test_skl_classifier
Scikit-learn classifier
tests/python/test_sklearn_integration.py:110
Function
test_skl_converter_iforest
Scikit-learn isolation forest
tests/python/test_sklearn_integration.py:180
Function
test_skl_converter_iforest_feature_subsampling
Scikit-learn isolation forest with feature subsampling
tests/python/test_sklearn_integration.py:203
Function
test_skl_export_rf_classifier
Round trip with scikit-learn RF classifier
tests/python/test_sklearn_integration.py:315
Function
test_skl_export_rf_multitarget_multiclass
Round trip with scikit-learn RF classifier, with multiple outputs and classes
tests/python/test_sklearn_integration.py:334
Function
test_skl_export_rf_regressor
Round trip with scikit-learn RF regressor
tests/python/test_sklearn_integration.py:293
Function
test_skl_hist_gradient_boosting_with_categorical
Scikit-learn HistGradientBoostingClassifier, with categorical splits
tests/python/test_sklearn_integration.py:231
Function
test_skl_hist_gradient_boosting_with_string_categorical
Scikit-learn HistGradientBoostingClassifier, with string categorical features
tests/python/test_sklearn_integration.py:267
Function
test_skl_multitarget_multiclass_rf
Scikit-learn RF classifier with multiple outputs and classes
tests/python/test_sklearn_integration.py:153
Function
test_skl_regressor
Scikit-learn regressor
tests/python/test_sklearn_integration.py:62
Function
test_tree_editing
Test ability to edit trees
tests/python/test_field_accessor.py:132
Function
test_xgb_dart
Test XGBoost DART model with dummy data
tests/python/test_xgboost_integration.py:293
Function
test_xgb_detect_format
Test functionality to detect the format of XGBoost (JSON vs. UBJSON)
tests/python/test_xgboost_integration.py:563
Function
test_xgb_multi_target_binary_classifier
Test XGBoost with multi-target binary classification problem
tests/python/test_xgboost_integration.py:425
Function
test_xgb_multi_target_regressor
Test XGBoost with multi-target regression problem
tests/python/test_xgboost_integration.py:496
Function
test_xgb_multiclass_classifier
Test XGBoost with multi-class classification problem
tests/python/test_xgboost_integration.py:146
Function
test_xgb_nonlinear_objective
Test XGBoost with non-linear objectives with synthetic data
tests/python/test_xgboost_integration.py:227
Function
test_xgb_regressor
Test XGBoost with regression data
tests/python/test_xgboost_integration.py:77
Function
throw_if
Throw an error/warning if there are positional arguments after the asterisk. Parameters ---------- func : functio
python/treelite/util.py:97
Method
~ContiguousArray
include/treelite/detail/contiguous_array.h:24
Method
~LogMessage
include/treelite/logging.h:140
Method
~LogMessageFatal
include/treelite/logging.h:126
Method
~LogMessageWarning
include/treelite/logging.h:157
Method
~Model
include/treelite/tree.h:461
Method
~ModelBuilder
include/treelite/model_builder.h:149
Method
~ModelPreset
include/treelite/tree.h:417
Method
~Tree
include/treelite/tree.h:89
← previous
601–686 of 686, ranked by callers