Skip to content

Commit 0d2acb1

Browse files
committed
documentation: Mark baseline as optional in KernelSHAP.
1 parent 3261dfa commit 0d2acb1

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/sagemaker/clarify.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -305,10 +305,11 @@ def __init__(
305305
"""Initializes config for SHAP.
306306
307307
Args:
308-
baseline (str or list): A list of rows (at least one) or S3 object URI to be used as
308+
baseline (None or str or list): None or S3 object Uri or A list of rows (at least one) to be used as
309309
the baseline dataset in the Kernel SHAP algorithm. The format should be the same
310310
as the dataset format. Each row should contain only the feature columns/values
311-
and omit the label column/values.
311+
and omit the label column/values. If None a baseline will be calculated automatically by using K-means
312+
or K-prototypes in the input dataset.
312313
num_samples (int): Number of samples to be used in the Kernel SHAP algorithm.
313314
This number determines the size of the generated synthetic dataset to compute the
314315
SHAP values.

0 commit comments

Comments
 (0)