MCPcopy Create free account

hub / github.com/dmlc/treelite / functions

Functions686 in github.com/dmlc/treelite

Functionload_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
Functionlogarithm_one_plus_exp
src/gtil/postprocessor.cc:50
Functionmain
tests/cpp/test_main.cc:9
Functionmain
tests/example_app/example.cc:17
Functionmain
tests/example_app/example.c:68
Methodmake_shared_enabler
src/model_loader/detail/xgboost_json/delegated_handler.cc:207
Functionmulticlass_ova
src/gtil/postprocessor.cc:78
Methodnull
src/model_loader/detail/xgboost_json/sax_adapters.cc:84
Methodnum_feature
Number of features used in the model
python/treelite/model.py:49
Methodnum_tree
Number of decision trees in the model
python/treelite/model.py:40
Methodnumber_float
src/model_loader/detail/xgboost_json/sax_adapters.cc:100
Methodnumber_integer
src/model_loader/detail/xgboost_json/sax_adapters.cc:92
Methodnumber_unsigned
src/model_loader/detail/xgboost_json/sax_adapters.cc:96
Methodoperator=
include/treelite/logging.h:121
Methodoperator==
include/treelite/detail/contiguous_array.h:251
Methodoutput_type
Output type
python/treelite/model.py:67
Methodparent
src/model_loader/xgboost_legacy.cc:229
Methodparse_error
src/model_loader/detail/xgboost_json/sax_adapters.cc:137
Methodpop_handler
src/model_loader/detail/xgboost_json/delegated_handler.cc:143
Functionpredict
Predict with a Treelite model using the General Tree Inference Library (GTIL). Parameters ---------- model : :py:class:`Model` objec
python/treelite/gtil/gtil.py:54
Functionpredict_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
Functionpredict_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
Methodpush_handler
src/model_loader/detail/xgboost_json/delegated_handler.h:135
Methodpush_key_handler
src/model_loader/detail/xgboost_json/delegated_handler.h:150
Methodset_cur_key
src/model_loader/detail/xgboost_json/delegated_handler.cc:152
Methodset_field
Set a field Parameters ---------- name: Name of the field. Consult :doc:`the model spec </serialization/
python/treelite/model.py:477
Functionsetup
(app)
docs/conf.py:228
Functionshow_progress
Show file download progress.
dev/prepare_pypi_release.py:24
Functionsigmoid
src/gtil/postprocessor.cc:34
Functionsigned_square
src/gtil/postprocessor.cc:23
Functionstandard_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
Functionstandard_settings
Default hypothesis settings
tests/python/hypothesis_util.py:324
Methodstart_array
src/model_loader/detail/xgboost_json/sax_adapters.cc:125
Methodstart_object
src/model_loader/detail/xgboost_json/sax_adapters.cc:117
Methodstore_int
src/model_loader/detail/xgboost_json/delegated_handler.h:312
Methodstore_string
src/model_loader/detail/xgboost_json/delegated_handler.h:349
Methodsum_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
Functiontest_categorical_encoding
Treelite should throw an exception when XGBoost model uses categorical encoding
tests/python/test_xgboost_integration.py:611
Functiontest_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
Functiontest_getter_setter
Test getter and setter methods
tests/python/test_field_accessor.py:15
Functiontest_invalid_node_id
Test for invalid node IDs
tests/python/test_model_builder.py:44
Functiontest_leaf_vector_rf
Test a small random forest with leaf vector output
tests/python/test_model_builder.py:89
Functiontest_lightgbm_binary_classification
Test LightGBM binary classifier
tests/python/test_lightgbm_integration.py:101
Functiontest_lightgbm_categorical_data
Test LightGBM with toy categorical data
tests/python/test_lightgbm_integration.py:209
Functiontest_lightgbm_deep_tree
Test LightGBM model with depth 32+
tests/python/test_lightgbm_integration.py:272
Functiontest_lightgbm_multiclass_classification
Test LightGBM multi-class classifier
tests/python/test_lightgbm_integration.py:158
Functiontest_lightgbm_regression
Test LightGBM regressor
tests/python/test_lightgbm_integration.py:46
Functiontest_lightgbm_sparse_categorical_model
Test LightGBM with high-cardinality categorical features
tests/python/test_lightgbm_integration.py:252
Functiontest_lightgbm_sparse_ranking_model
Generate a LightGBM ranking model with highly sparse data.
tests/python/test_lightgbm_integration.py:221
Functiontest_load_old_xgboost_model
Ensure that Treelite can load old XGBoost models
tests/python/test_xgboost_integration.py:598
Functiontest_model_concat_with_mismatched_tree_types
Test model concatenation with a tree stump
tests/python/test_model_concat.py:91
Functiontest_model_concat_with_tree_stump
Test model concatenation with a tree stump
tests/python/test_model_concat.py:18
Functiontest_orphaned_nodes
Test for orphaned nodes
tests/python/test_model_builder.py:16
Functiontest_predict_special_with_binary_classifier
Test predict_leaf / predict_per_tree with XGBoost binary classifier
tests/python/test_gtil.py:77
Functiontest_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
Functiontest_predict_special_with_regressor
Test predict_leaf / predict_per_tree with XGBoost regressor
tests/python/test_gtil.py:37
Functiontest_serialize_as_buffer
Test whether Treelite objects can be serialized to a buffer
tests/serializer/test_serializer.py:54
Functiontest_serialize_as_bytes
Test whether Treelite objects can be serialized to a byte sequence
tests/python/test_serializer.py:58
Functiontest_serialize_as_checkpoint
Test whether Treelite objects can be serialized to a checkpoint
tests/python/test_serializer.py:112
Functiontest_skl_classifier
Scikit-learn classifier
tests/python/test_sklearn_integration.py:110
Functiontest_skl_converter_iforest
Scikit-learn isolation forest
tests/python/test_sklearn_integration.py:180
Functiontest_skl_converter_iforest_feature_subsampling
Scikit-learn isolation forest with feature subsampling
tests/python/test_sklearn_integration.py:203
Functiontest_skl_export_rf_classifier
Round trip with scikit-learn RF classifier
tests/python/test_sklearn_integration.py:315
Functiontest_skl_export_rf_multitarget_multiclass
Round trip with scikit-learn RF classifier, with multiple outputs and classes
tests/python/test_sklearn_integration.py:334
Functiontest_skl_export_rf_regressor
Round trip with scikit-learn RF regressor
tests/python/test_sklearn_integration.py:293
Functiontest_skl_hist_gradient_boosting_with_categorical
Scikit-learn HistGradientBoostingClassifier, with categorical splits
tests/python/test_sklearn_integration.py:231
Functiontest_skl_hist_gradient_boosting_with_string_categorical
Scikit-learn HistGradientBoostingClassifier, with string categorical features
tests/python/test_sklearn_integration.py:267
Functiontest_skl_multitarget_multiclass_rf
Scikit-learn RF classifier with multiple outputs and classes
tests/python/test_sklearn_integration.py:153
Functiontest_skl_regressor
Scikit-learn regressor
tests/python/test_sklearn_integration.py:62
Functiontest_tree_editing
Test ability to edit trees
tests/python/test_field_accessor.py:132
Functiontest_xgb_dart
Test XGBoost DART model with dummy data
tests/python/test_xgboost_integration.py:293
Functiontest_xgb_detect_format
Test functionality to detect the format of XGBoost (JSON vs. UBJSON)
tests/python/test_xgboost_integration.py:563
Functiontest_xgb_multi_target_binary_classifier
Test XGBoost with multi-target binary classification problem
tests/python/test_xgboost_integration.py:425
Functiontest_xgb_multi_target_regressor
Test XGBoost with multi-target regression problem
tests/python/test_xgboost_integration.py:496
Functiontest_xgb_multiclass_classifier
Test XGBoost with multi-class classification problem
tests/python/test_xgboost_integration.py:146
Functiontest_xgb_nonlinear_objective
Test XGBoost with non-linear objectives with synthetic data
tests/python/test_xgboost_integration.py:227
Functiontest_xgb_regressor
Test XGBoost with regression data
tests/python/test_xgboost_integration.py:77
Functionthrow_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
← previous601–686 of 686, ranked by callers