diff --git a/shapiq/approximator/regression/fsi.py b/shapiq/approximator/regression/fsi.py index 3384b450..a0d888ad 100644 --- a/shapiq/approximator/regression/fsi.py +++ b/shapiq/approximator/regression/fsi.py @@ -6,7 +6,7 @@ class RegressionFSI(Regression, NShapleyMixin): - """Estimates the FSI values using the weighted least square approach. + """Estimates the FSI values [1] using the weighted least square approach. Args: n: The number of players. @@ -20,6 +20,11 @@ class RegressionFSI(Regression, NShapleyMixin): min_order: The minimum order of the approximation. For FSI, min_order is equal to 1. iteration_cost: The cost of a single iteration of the regression FSI. + References: + [1]: Tsai, C.-P., Yeh, C.-K., & Ravikumar, P. (2023). Faith-Shap: The Faithful Shapley + Interaction Index. J. Mach. Learn. Res., 24, 94:1-94:42. Retrieved from + http://jmlr.org/papers/v24/22-0202.html + Example: >>> from games import DummyGame >>> from approximator import RegressionFSI