Skip to content

Commit 5c26c4e

Browse files
shinglyuShing Lyu
and
Shing Lyu
authored
documentation: fix the ClarifyCheckStep documentation to mention PDP (#4259)
* documentation: fix the ClarifyCheckStep documentation to mention PDP support * fix: break the lines to meet pylint requirement --------- Co-authored-by: Shing Lyu <[email protected]>
1 parent e75865f commit 5c26c4e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sagemaker/workflow/clarify_check_step.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,10 @@ class ModelExplainabilityCheckConfig(ClarifyCheckConfig):
132132
133133
Attributes:
134134
model_config (ModelConfig): Config of the model and its endpoint to be created.
135-
explainability_config (SHAPConfig): Config of the specific explainability method.
136-
Currently, only SHAP is supported.
135+
explainability_config (SHAPConfig or PDPConfig): Config of the explainability method.
136+
Supports SHAP or PDP.
137+
For `PDPConfig`, `features` must be specified.
138+
`top_k_features` based on SHAP is currently not supported.
137139
model_scores (str or int or ModelPredictedLabelConfig): Index or JMESPath expression
138140
to locate the predicted scores in the model output (default: None).
139141
This is not required if the model output is a single score. Alternatively,

0 commit comments

Comments
 (0)