Skip to content

Commit 43901a1

Browse files
authored
Update how-to-machine-learning-interpretability-automl.md
1 parent 103e831 commit 43901a1

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

articles/machine-learning/how-to-machine-learning-interpretability-automl.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -80,20 +80,21 @@ To generate an explanation for AutoML models, use the `MimicWrapper` class. You
8080

8181
- The explainer setup object
8282
- Your workspace
83-
- A LightGBM model, which acts as a surrogate to the `fitted_model` automated ML model
83+
- A surrogate model to explain the `fitted_model` automated ML model
8484

8585
The MimicWrapper also takes the `automl_run` object where the engineered explanations will be uploaded.
8686

8787
```python
88-
from azureml.explain.model.mimic.models.lightgbm_model import LGBMExplainableModel
8988
from azureml.explain.model.mimic_wrapper import MimicWrapper
9089

9190
# Initialize the Mimic Explainer
92-
explainer = MimicWrapper(ws, automl_explainer_setup_obj.automl_estimator, LGBMExplainableModel,
91+
explainer = MimicWrapper(ws, automl_explainer_setup_obj.automl_estimator,
92+
explainable_model=automl_explainer_setup_obj.surrogate_model,
9393
init_dataset=automl_explainer_setup_obj.X_transform, run=automl_run,
9494
features=automl_explainer_setup_obj.engineered_feature_names,
9595
feature_maps=[automl_explainer_setup_obj.feature_map],
96-
classes=automl_explainer_setup_obj.classes)
96+
classes=automl_explainer_setup_obj.classes,
97+
explainer_kwargs=automl_explainer_setup_obj.surrogate_model_params)
9798
```
9899

99100
### Use MimicExplainer for computing and visualizing engineered feature importance

0 commit comments

Comments
 (0)