diff --git a/dice_ml/explainer_interfaces/explainer_base.py b/dice_ml/explainer_interfaces/explainer_base.py index 59894f1f..756c7675 100644 --- a/dice_ml/explainer_interfaces/explainer_base.py +++ b/dice_ml/explainer_interfaces/explainer_base.py @@ -810,7 +810,7 @@ def get_model_output_from_scores(self, model_scores): else: # 1-D input model_output[i] = np.round(model_scores[i]) elif self.model.model_type == ModelTypes.Regressor: - model_output[i] = model_scores[i] + model_output[i] = float(np.squeeze(model_scores[i])) return model_output def check_permitted_range(self, permitted_range):