tree_shap.hpp File Reference

#include <cuml/ensemble/treelite_defs.hpp>
#include <cstddef>
#include <cstdint>
#include <memory>
#include <variant>

Include dependency graph for tree_shap.hpp:

Go to the source code of this file.

Namespaces
	ML
	ML::Explainer

Typedefs
using	ML::Explainer::TreePathHandle = std::variant< std::shared_ptr< TreePathInfo< float > >, std::shared_ptr< TreePathInfo< double > >>
using	ML::Explainer::FloatPointer = std::variant< float *, double * >

Functions
TreePathHandle	ML::Explainer::extract_path_info (TreeliteModelHandle model)
void	ML::Explainer::gpu_treeshap (TreePathHandle path_info, const FloatPointer data, std::size_t n_rows, std::size_t n_cols, FloatPointer out_preds, std::size_t out_preds_size)
void	ML::Explainer::gpu_treeshap_interventional (TreePathHandle path_info, const FloatPointer data, std::size_t n_rows, std::size_t n_cols, const FloatPointer background_data, std::size_t background_n_rows, std::size_t background_n_cols, FloatPointer out_preds, std::size_t out_preds_size)
void	ML::Explainer::gpu_treeshap_interactions (TreePathHandle path_info, const FloatPointer data, std::size_t n_rows, std::size_t n_cols, FloatPointer out_preds, std::size_t out_preds_size)
void	ML::Explainer::gpu_treeshap_taylor_interactions (TreePathHandle path_info, const FloatPointer data, std::size_t n_rows, std::size_t n_cols, FloatPointer out_preds, std::size_t out_preds_size)