wehub-resource-sync
14 lines
925 B
C++
14 lines
925 B
C++
// see https://nanobind.readthedocs.io/en/latest/basics.html#basics and following docs
|
|
#include <nanobind/nanobind.h>
|
|
#include "grey_code_utils.h"
|
|
#include "kernel_explainer_utils.h"
|
|
|
|
namespace nb = nanobind;
|
|
|
|
NB_MODULE(_cutils, m)
|
|
{
|
|
m.def("compute_grey_code_row_values", &compute_grey_code_row_values_1d, "row_values"_a, "mask"_a, "inds"_a, "outputs"_a, "shapley_coeff"_a, "extended_delta_indexes"_a, "noop_code"_a, "Compute the row values for the grey code algorithm in 1D");
|
|
m.def("compute_grey_code_row_values", &compute_grey_code_row_values_2d, "row_values"_a, "mask"_a, "inds"_a, "outputs"_a, "shapley_coeff"_a, "extended_delta_indexes"_a, "noop_code"_a, "Compute the row values for the grey code algorithm in 2D");
|
|
m.def("compute_exp_val", &compute_exp_val, "nsamples_run"_a, "nsamples_added"_a, "D"_a, "N"_a, "weights"_a, "y"_a, "ey"_a, "Compute the expected value for the kernel explainer algorithm");
|
|
}
|