Skip to content

Commit a7c3a0f

Browse files
committed
update samples from Release-54 as a part of SDK release
1 parent 6d11cdf commit a7c3a0f

File tree

1 file changed

+4
-4
lines changed
  • how-to-use-azureml/automated-machine-learning/regression-explanation-featurization

1 file changed

+4
-4
lines changed

how-to-use-azureml/automated-machine-learning/regression-explanation-featurization/train_explainer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,14 @@
44

55
from azureml.core.run import Run
66
from azureml.core.experiment import Experiment
7-
from sklearn.externals import joblib
87
from azureml.core.dataset import Dataset
98
from azureml.train.automl.runtime.automl_explain_utilities import AutoMLExplainerSetupClass, \
109
automl_setup_model_explanations, automl_check_model_if_explainable
1110
from azureml.explain.model.mimic.models.lightgbm_model import LGBMExplainableModel
1211
from azureml.explain.model.mimic_wrapper import MimicWrapper
1312
from azureml.automl.core.shared.constants import MODEL_PATH
14-
from azureml.explain.model.scoring.scoring_explainer import TreeScoringExplainer, save
15-
13+
from azureml.explain.model.scoring.scoring_explainer import TreeScoringExplainer
14+
import joblib
1615

1716
OUTPUT_DIR = './outputs/'
1817
os.makedirs(OUTPUT_DIR, exist_ok=True)
@@ -74,7 +73,8 @@
7473
scoring_explainer = TreeScoringExplainer(explainer.explainer, feature_maps=[automl_explainer_setup_obj.feature_map])
7574

7675
# Pickle scoring explainer locally
77-
save(scoring_explainer, exist_ok=True)
76+
with open('scoring_explainer.pkl', 'wb') as stream:
77+
joblib.dump(scoring_explainer, stream)
7878

7979
# Upload the scoring explainer to the automl run
8080
automl_run.upload_file('outputs/scoring_explainer.pkl', 'scoring_explainer.pkl')

0 commit comments

Comments
 (0)