Skip to content

Commit 161c6ca

Browse files
shinglyuakrishna1995
authored andcommitted
documentation: fix the ClarifyCheckStep documentation to mention PDP support
1 parent c797f2d commit 161c6ca

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/sagemaker/workflow/clarify_check_step.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,9 @@ 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 specific explainability method.
136+
Supports SHAP or PDP.
137+
For `PDPConfig`, `features` must be specified. `top_k_features` based on SHAP is currently not supported.
137138
model_scores (str or int or ModelPredictedLabelConfig): Index or JMESPath expression
138139
to locate the predicted scores in the model output (default: None).
139140
This is not required if the model output is a single score. Alternatively,

0 commit comments

Comments
 (0)