diff --git a/hiclass/Explainer.py b/hiclass/Explainer.py index 78425a71..98d23f25 100644 --- a/hiclass/Explainer.py +++ b/hiclass/Explainer.py @@ -49,7 +49,7 @@ def __init__( n_jobs : int, default=1 The number of jobs to run in parallel. algorithm : str, default="auto" - The algorithm to use for SHAP explainer. Possible values are 'linear', 'tree', 'auto', 'permutation'. or 'partition' + The algorithm to use for SHAP explainer. Possible values are 'linear', 'tree', 'auto', 'permutation' or 'partition' mode : str, default="" The mode of the SHAP explainer. Can be 'tree', 'gradient', 'deep', 'linear', or '' for default SHAP explainer.