File tree 1 file changed +3
-2
lines changed
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -292,10 +292,11 @@ def __init__(
292
292
"""Initializes config for SHAP.
293
293
294
294
Args:
295
- baseline (str or list): A list of rows (at least one) or S3 object URI to be used as
295
+ baseline (None or str or list): None or S3 object Uri or A list of rows (at least one) to be used as
296
296
the baseline dataset in the Kernel SHAP algorithm. The format should be the same
297
297
as the dataset format. Each row should contain only the feature columns/values
298
- and omit the label column/values.
298
+ and omit the label column/values. If None a baseline will be calculated automatically by using K-means
299
+ or K-prototypes in the input dataset.
299
300
num_samples (int): Number of samples to be used in the Kernel SHAP algorithm.
300
301
This number determines the size of the generated synthetic dataset to compute the
301
302
SHAP values.
You can’t perform that action at this time.
0 commit comments